Jyothirmai commited on
Commit
8feeeb8
·
verified ·
1 Parent(s): 1992bc2

Update tester.py

Browse files
Files changed (1) hide show
  1. tester.py +3 -3
tester.py CHANGED
@@ -81,9 +81,9 @@ class CaptionSampler(object):
81
  self.sentence_model.eval()
82
  self.word_model.eval()
83
 
84
- imageData = image_file.convert('RGB')
85
- imageData = self.transform(imageData)
86
- imageData = imageData.unsqueeze_(0)
87
 
88
  print(imageData.shape)
89
 
 
81
  self.sentence_model.eval()
82
  self.word_model.eval()
83
 
84
+ # imageData = image_file.convert('RGB')
85
+ # imageData = self.transform(imageData)
86
+ imageData = image_file.unsqueeze_(0)
87
 
88
  print(imageData.shape)
89