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

Update tester.py

Browse files
Files changed (1) hide show
  1. tester.py +2 -2
tester.py CHANGED
@@ -81,8 +81,8 @@ class CaptionSampler(object):
81
  self.sentence_model.eval()
82
  self.word_model.eval()
83
 
84
- # imageData = Image.open(image_file).convert('RGB')
85
- imageData = self.transform(image_file)
86
  imageData = imageData.unsqueeze_(0)
87
 
88
  print(imageData.shape)
 
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)