ENSEEIHT/projet-long
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Fix import in utils

Browse source 
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
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
predict.py
View file

@ -10,8 +10,7 @@ from torchvision import transforms
from utils.data_loading import BasicDataset from utils.data_loading import BasicDataset
from unet import UNet from unet import UNet
from utils import plot_img_and_mask from utils.utils import plot_img_and_mask
def predict_img(net, def predict_img(net,
full_img, full_img,