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

Convert to tensor in predict.py

Browse source 
Former-commit-id: be1ad32304f2365c0c1db6b0e7fb835acc0fbfed
This commit is contained in:
milesial 2019-12-04 12:43:04 +01:00 committed by GitHub
parent 5a7e934560
commit 0d1cc25ae2
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
predict.py
View file

@ -23,7 +23,7 @@ def predict_img(net,
net.eval() net.eval()
ds = BasicDataset('', '', scale=scale_factor) ds = BasicDataset('', '', scale=scale_factor)
img = ds.preprocess(full_img) img = torch.from_numpy(ds.preprocess(full_img))
img = img.unsqueeze(0) img = img.unsqueeze(0)
img = img.to(device=device, dtype=torch.float32) img = img.to(device=device, dtype=torch.float32)