From d1083513f9b10ce32f9a71ecdcaffb9203ecac75 Mon Sep 17 00:00:00 2001 From: Your Name Date: Mon, 27 Jun 2022 16:41:03 +0200 Subject: [PATCH] refactor: renamed a file Former-commit-id: b1e3f616bbadd8087ad52b13152d4dc72d1267aa --- src/evaluate.py | 2 +- src/train.py | 2 +- src/utils/{dice_score.py => dice.py} | 0 3 files changed, 2 insertions(+), 2 deletions(-) rename src/utils/{dice_score.py => dice.py} (100%) diff --git a/src/evaluate.py b/src/evaluate.py index c67a60b..bdaeb78 100644 --- a/src/evaluate.py +++ b/src/evaluate.py @@ -2,7 +2,7 @@ import torch import torch.nn.functional as F from tqdm import tqdm -from utils.dice_score import multiclass_dice_coeff, dice_coeff +from src.utils.dice import multiclass_dice_coeff, dice_coeff def evaluate(net, dataloader, device): diff --git a/src/train.py b/src/train.py index 3e559a7..690273e 100644 --- a/src/train.py +++ b/src/train.py @@ -13,7 +13,7 @@ from tqdm import tqdm from evaluate import evaluate from src.utils.dataset import BasicDataset, CarvanaDataset from unet import UNet -from utils.dice_score import dice_loss +from src.utils.dice import dice_loss dir_img = Path("./data/imgs/") dir_mask = Path("./data/masks/") diff --git a/src/utils/dice_score.py b/src/utils/dice.py similarity index 100% rename from src/utils/dice_score.py rename to src/utils/dice.py