refactor: renamed files

Former-commit-id: c9936db9c051268c7e9bd3a78bbeaab9948e416e

src/predict.py

@@ -8,7 +8,7 @@ import torch.nn.functional as F
 from PIL import Image
 from torchvision import transforms
 
-from utils.data_loading import BasicDataset
+from src.utils.dataset import BasicDataset
 from unet import UNet
 from utils.utils import plot_img_and_mask
 

src/train.py

@@ -11,8 +11,8 @@ from torch.utils.data import DataLoader, random_split
 from tqdm import tqdm
 
 from evaluate import evaluate
+from src.utils.dataset import BasicDataset, CarvanaDataset
 from unet import UNet
-from utils.data_loading import BasicDataset, CarvanaDataset
 from utils.dice_score import dice_loss
 
 dir_img = Path("./data/imgs/")

src/unet/__init__.py

@@ -1 +1 @@
-from .unet_model import UNet
+from .model import UNet

src/unet/model.py

@@ -1,6 +1,6 @@
 """ Full assembly of the parts to form the complete network """
 
-from .unet_parts import *
+from .blocks import *
 
 
 class UNet(nn.Module):