Fix import in utils

Former-commit-id: faa6e13ea9b53d6773e465a2a60617ffc3d27939
This commit is contained in:
milesial 2021-08-19 09:40:54 +02:00 committed by GitHub
parent 4e638b47de
commit ce4567c908

View file

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