未验证 提交 bd29c34f 编写于 作者: S SunGaofeng 提交者: GitHub

fix pickle.load in ctcn_reader for python3 (#2679)

fix pickle.load to fit both python3 and 2
上级 90acc7c1
...@@ -199,18 +199,18 @@ class CTCNReader(DataReader): ...@@ -199,18 +199,18 @@ class CTCNReader(DataReader):
def load_file(self, fname): def load_file(self, fname):
if python_ver < (3, 0): if python_ver < (3, 0):
rgb_pkl = pickle.load( 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( 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: else:
rgb_pkl = pickle.load( 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') encoding='bytes')
flow_pkl = pickle.load( 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') encoding='bytes')
data_flow = np.array(flow_pkl['scores']) data_flow = np.array(flow_pkl[b'scores'])
data_rgb = np.array(rgb_pkl['scores']) data_rgb = np.array(rgb_pkl[b'scores'])
if data_flow.shape[0] < data_rgb.shape[0]: if data_flow.shape[0] < data_rgb.shape[0]:
data_rgb = data_rgb[0:data_flow.shape[0], :] data_rgb = data_rgb[0:data_flow.shape[0], :]
elif data_flow.shape[0] > data_rgb.shape[0]: elif data_flow.shape[0] > data_rgb.shape[0]:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册