From 524a0692280e5402d6732751e8d5286afca19448 Mon Sep 17 00:00:00 2001 From: SunGaofeng Date: Wed, 4 Sep 2019 20:35:34 -0500 Subject: [PATCH] fix data transfer of youtube8m and kinetics to fit for python3 (#3268) --- PaddleCV/PaddleVideo/data/dataset/kinetics/video2pkl.py | 2 +- PaddleCV/PaddleVideo/data/dataset/youtube8m/tf2pkl.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/PaddleCV/PaddleVideo/data/dataset/kinetics/video2pkl.py b/PaddleCV/PaddleVideo/data/dataset/kinetics/video2pkl.py index 0950b01b..a97ae053 100644 --- a/PaddleCV/PaddleVideo/data/dataset/kinetics/video2pkl.py +++ b/PaddleCV/PaddleVideo/data/dataset/kinetics/video2pkl.py @@ -68,7 +68,7 @@ def generate_pkl(entry): images = sorted(glob.glob(vid + '/*.jpg')) ims = [] for img in images: - f = open(img) + f = open(img, 'rb') ims.append(f.read()) f.close() diff --git a/PaddleCV/PaddleVideo/data/dataset/youtube8m/tf2pkl.py b/PaddleCV/PaddleVideo/data/dataset/youtube8m/tf2pkl.py index 7184850b..5a559da1 100644 --- a/PaddleCV/PaddleVideo/data/dataset/youtube8m/tf2pkl.py +++ b/PaddleCV/PaddleVideo/data/dataset/youtube8m/tf2pkl.py @@ -263,7 +263,7 @@ def main(files_pattern): coord.request_stop() coord.join(threads) - print vid_num + print(vid_num) record_name = files_pattern.split('/')[-1].split('.')[0] outputdir = target_dir -- GitLab