From bd29c34feeab7c26d309da7b560fe1587113cfe4 Mon Sep 17 00:00:00 2001 From: SunGaofeng Date: Tue, 2 Jul 2019 20:12:34 +0800 Subject: [PATCH] fix pickle.load in ctcn_reader for python3 (#2679) fix pickle.load to fit both python3 and 2 --- PaddleCV/PaddleVideo/datareader/ctcn_reader.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/PaddleCV/PaddleVideo/datareader/ctcn_reader.py b/PaddleCV/PaddleVideo/datareader/ctcn_reader.py index 174f7696..e5fe8ee7 100644 --- a/PaddleCV/PaddleVideo/datareader/ctcn_reader.py +++ b/PaddleCV/PaddleVideo/datareader/ctcn_reader.py @@ -199,18 +199,18 @@ class CTCNReader(DataReader): def load_file(self, fname): if python_ver < (3, 0): rgb_pkl = pickle.load( - open(os.path.join(self.root, self.rgb, fname + '.pkl'))) + open(os.path.join(self.root, self.rgb, fname + '.pkl'), 'rb')) flow_pkl = pickle.load( - open(os.path.join(self.root, self.flow, fname + '.pkl'))) + open(os.path.join(self.root, self.flow, fname + '.pkl'), 'rb')) else: rgb_pkl = pickle.load( - open(os.path.join(self.root, self.rgb, fname + '.pkl')), + open(os.path.join(self.root, self.rgb, fname + '.pkl'), 'rb'), encoding='bytes') flow_pkl = pickle.load( - open(os.path.join(self.root, self.flow, fname + '.pkl')), + open(os.path.join(self.root, self.flow, fname + '.pkl'), 'rb'), encoding='bytes') - data_flow = np.array(flow_pkl['scores']) - data_rgb = np.array(rgb_pkl['scores']) + data_flow = np.array(flow_pkl[b'scores']) + data_rgb = np.array(rgb_pkl[b'scores']) if data_flow.shape[0] < data_rgb.shape[0]: data_rgb = data_rgb[0:data_flow.shape[0], :] elif data_flow.shape[0] > data_rgb.shape[0]: -- GitLab