Jyothirmai commited on
Commit
c8b938d
1 Parent(s): 05fd4a4

Update tester.py

Browse files
Files changed (1) hide show
  1. tester.py +2 -7
tester.py CHANGED
@@ -81,13 +81,8 @@ class CaptionSampler(object):
81
  self.sentence_model.eval()
82
  self.word_model.eval()
83
 
84
- transform = transforms.Compose([transforms.ToTensor()])
85
-
86
- imageData = transform(image_file)
87
- print(imageData)
88
-
89
- # imageData = imageData.convert('RGB')
90
- imageData = self.transform(imageData)
91
  imageData = imageData.unsqueeze_(0)
92
 
93
 
 
81
  self.sentence_model.eval()
82
  self.word_model.eval()
83
 
84
+
85
+ imageData = self.transform(image_file)
 
 
 
 
 
86
  imageData = imageData.unsqueeze_(0)
87
 
88