提交 7e88dd2a 编写于 作者: S SunGaofeng

modify viedo2pkl.py to be consistent with kinetics reader

上级 86c583f1
...@@ -72,7 +72,7 @@ def generate_pkl(entry): ...@@ -72,7 +72,7 @@ def generate_pkl(entry):
output_pkl = vid + ".pkl" output_pkl = vid + ".pkl"
output_pkl = os.path.join(target_dir, output_pkl) output_pkl = os.path.join(target_dir, output_pkl)
f = open(output_pkl, 'w') f = open(output_pkl, 'w')
cPickle.dump((vid, ims, label), f, -1) cPickle.dump((vid, label, ims), f, -1)
f.close() f.close()
os.system('rm -rf %s' % vid) os.system('rm -rf %s' % vid)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册