提交 0b48d1cc 编写于 作者: Q qipengfei7

new robust reader

上级 649c536b
...@@ -351,7 +351,7 @@ class Reader(object): ...@@ -351,7 +351,7 @@ class Reader(object):
if self._load_img: if self._load_img:
sample['mixup']['image'] = self._load_image(sample['mixup'][ sample['mixup']['image'] = self._load_image(sample['mixup'][
'im_file']) 'im_file'])
if is_valid_path: if is_valid_img_path:
batch.append(sample) batch.append(sample)
bs += 1 bs += 1
return batch return batch
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册