未验证 提交 188536e0 编写于 作者: M minghaoBD 提交者: GitHub

fix bug in imagenet reader (#1284)

上级 94002d7b
......@@ -143,6 +143,8 @@ def _reader_creator(file_list,
color_jitter=False,
rotate=False,
data_dir=DATA_DIR,
crop_size=DATA_DIM,
resize_size=RESIZE_DIM,
batch_size=1):
def reader():
try:
......@@ -164,7 +166,12 @@ def _reader_creator(file_list,
os._exit(1)
mapper = functools.partial(
process_image, mode=mode, color_jitter=color_jitter, rotate=rotate)
process_image,
mode=mode,
color_jitter=color_jitter,
rotate=rotate,
crop_size=crop_size,
resize_size=resize_size)
return paddle.reader.xmap_readers(mapper, reader, THREAD, BUF_SIZE)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册