From 84c3b2b200834c0d1ec36bc64254008fe03c5e99 Mon Sep 17 00:00:00 2001 From: Anujraaj Goyal Date: Tue, 11 Feb 2020 19:26:45 +0800 Subject: [PATCH] Migration to Py3 for MSRVTT dataset --- MSRVTT.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/MSRVTT.py b/MSRVTT.py index b328840..be6a45c 100644 --- a/MSRVTT.py +++ b/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)