zamborg commited on
Commit
1df8b5f
1 Parent(s): 09da12b

syntax fix

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -50,7 +50,7 @@ class ImageLoader():
50
  x,y = image.shape[-2:]
51
  ratio = float(self.show_size/max((x,y)))
52
  image = torchvision.transforms.functional.resize(image, [int(x * ratio), int(y * ratio)])
53
- return torch.transforms.functional.to_pil_image(image)
54
 
55
 
56
  class VirTexModel():
50
  x,y = image.shape[-2:]
51
  ratio = float(self.show_size/max((x,y)))
52
  image = torchvision.transforms.functional.resize(image, [int(x * ratio), int(y * ratio)])
53
+ return torchvision.transforms.functional.to_pil_image(image)
54
 
55
 
56
  class VirTexModel():