Restructure

Former-commit-id: beadb49b75ea79a3c0f95df589f64a8274419c5b
This commit is contained in:
milesial 2021-08-16 06:08:09 +02:00
parent a7270ca699
commit 5cd2a3b0b7
9 changed files with 5 additions and 5 deletions

View file

@ -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

View file

@ -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):

View file

@ -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

View file

@ -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
View file