Restructure
Former-commit-id: beadb49b75ea79a3c0f95df589f64a8274419c5b
This commit is contained in:
parent
a7270ca699
commit
5cd2a3b0b7
|
@ -109,7 +109,7 @@ The Carvana data is available on the [Kaggle website](https://www.kaggle.com/c/c
|
|||
You can also download it using your Kaggle API key with:
|
||||
|
||||
```shell script
|
||||
bash download_data.sh <username> <apikey>
|
||||
bash scripts/download_data.sh <username> <apikey>
|
||||
```
|
||||
|
||||
## Notes on memory
|
||||
|
|
|
@ -2,7 +2,7 @@ import torch
|
|||
import torch.nn.functional as F
|
||||
from tqdm import tqdm
|
||||
|
||||
from dice_score import multiclass_dice_coeff
|
||||
from utils.dice_score import multiclass_dice_coeff
|
||||
|
||||
|
||||
def evaluate(net, dataloader, device):
|
||||
|
|
|
@ -8,7 +8,7 @@ import torch.nn.functional as F
|
|||
from PIL import Image
|
||||
from torchvision import transforms
|
||||
|
||||
from data_loading import BasicDataset
|
||||
from utils.data_loading import BasicDataset
|
||||
from unet import UNet
|
||||
from utils import plot_img_and_mask
|
||||
|
||||
|
|
4
train.py
4
train.py
|
@ -11,8 +11,8 @@ from torch import optim
|
|||
from torch.utils.data import DataLoader, random_split
|
||||
from tqdm import tqdm
|
||||
|
||||
from data_loading import BasicDataset, CarvanaDataset
|
||||
from dice_score import dice_loss
|
||||
from utils.data_loading import BasicDataset, CarvanaDataset
|
||||
from utils.dice_score import dice_loss
|
||||
from evaluate import evaluate
|
||||
from unet import UNet
|
||||
|
||||
|
|
0
utils/__init__.py
Normal file
0
utils/__init__.py
Normal file
Loading…
Reference in a new issue