diff --git a/ppcls/data/reader.py b/ppcls/data/reader.py index 5749cac13baf8627585b8d0c10bcb86f1ab58144..e68ebd77e9d4f9c901be434e8248b57e7eb0127b 100755 --- a/ppcls/data/reader.py +++ b/ppcls/data/reader.py @@ -189,9 +189,9 @@ def partial_reader(params, full_lines, part_id=0, part_num=1): for line in full_lines: img_path, label = line.split() img_path = os.path.join(params['data_dir'], img_path) - img = open(img_path).read() - img = transform(img, ops) - yield (img, int(label)) + with open(img_path, 'rb') as f: + img = f.read() + yield (transform(img, ops), int(label)) return reader