zamborg commited on
Commit
214bd84
1 Parent(s): a8416ee
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -25,8 +25,8 @@ class ImageLoader():
25
  torchvision.transforms.ToTensor(),
26
  torchvision.transforms.Resize(256),
27
  torchvision.transforms.CenterCrop(224),
28
- torchvision.transforms.Normalize((.485, .456, .406), (.229, .224, .225)),
29
- torchvision.transforms.ToTensor()])
30
  def load(self, im_path):
31
  im = torch.FloatTensor(self.transformer(Image.open(im_path))).unsqueeze(0)
32
  return {"image": im}
25
  torchvision.transforms.ToTensor(),
26
  torchvision.transforms.Resize(256),
27
  torchvision.transforms.CenterCrop(224),
28
+ torchvision.transforms.Normalize((.485, .456, .406), (.229, .224, .225))])
29
+
30
  def load(self, im_path):
31
  im = torch.FloatTensor(self.transformer(Image.open(im_path))).unsqueeze(0)
32
  return {"image": im}