From ce4567c9083a3c3aaf1ae122f70fd91db761e65f Mon Sep 17 00:00:00 2001 From: milesial Date: Thu, 19 Aug 2021 09:40:54 +0200 Subject: [PATCH] Fix import in utils Former-commit-id: faa6e13ea9b53d6773e465a2a60617ffc3d27939 --- predict.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/predict.py b/predict.py index 296fdb8..fdb95ea 100755 --- a/predict.py +++ b/predict.py @@ -10,8 +10,7 @@ from torchvision import transforms from utils.data_loading import BasicDataset from unet import UNet -from utils import plot_img_and_mask - +from utils.utils import plot_img_and_mask def predict_img(net, full_img,