Migration to Py3 for MSRVTT dataset

MSRVTT.py

@@ -26,8 +26,8 @@ class MSRVTT(Dataset):
     """LSMDC dataset."""
 
     def __init__(self, visual_features, flow_features, text_features, audio_features,
-            face_features, train_list, test_list, coco_visual_path='data/X_train2014_resnet152.npy',
-            coco_text_path='data/w2v_coco_train2014_1.npy',coco=True, max_words=30,verbose=False):
+            face_features, train_list, test_list, coco_visual_path='data/data/X_train2014_resnet152.npy',
+            coco_text_path='data/data/w2v_coco_train2014_1.npy',coco=True, max_words=30,verbose=False):
         """
         Args:
         """
@@ -46,26 +46,26 @@ def __init__(self, visual_features, flow_features, text_features, audio_features
 
 
         pickle_in = open(visual_features,'rb')
-        self.visual_features = pickle.load(pickle_in)
+        self.visual_features = pickle.load(pickle_in, encoding='latin1')
 
         pickle_in = open(flow_features,'rb')
-        self.flow_features = pickle.load(pickle_in)
+        self.flow_features = pickle.load(pickle_in, encoding='latin1')
 
         pickle_in = open(audio_features,'rb')
-        self.audio_features = pickle.load(pickle_in)
+        self.audio_features = pickle.load(pickle_in, encoding='latin1')
 
         pickle_in = open(text_features,'rb')
-        self.text_features = pickle.load(pickle_in)
+        self.text_features = pickle.load(pickle_in, encoding='latin1')
 
         pickle_in = open(face_features,'rb')
-        self.face_features = pickle.load(pickle_in)
+        self.face_features = pickle.load(pickle_in, encoding='latin1')
 
         self.coco = coco
 
         if coco:
             # adding coco data
             self.coco_visual = np.load(coco_visual_path)
-            self.coco_text = np.load(coco_text_path)
+            self.coco_text = np.load(coco_text_path, encoding='latin1')
 
             self.n_MSR = len(self.train_list)
             self.n_coco = len(self.coco_visual)