Update tester.py

tester.py

@@ -80,8 +80,15 @@ class CaptionSampler(object):
         self.co_attention.eval()
         self.sentence_model.eval()
         self.word_model.eval()
-        
-        imageData = image_file.convert('RGB')
+
+        transform = transforms.Compose([ 
+        transforms.PILToTensor() 
+        ]) 
+       
+        imageData = transform(image_file) 
+        print(imageData) 
+            
+        # imageData = imageData.convert('RGB')
         imageData = self.transform(imageData)
         imageData = imageData.unsqueeze_(0)