new robust reader

0b48d1cc · qipengfei7 · 649c536b · 0b48d1cc
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppdet/data/reader.py ppdet/data/reader.py +1 -1

未找到文件。
--- a/ppdet/data/reader.py
+++ b/ppdet/data/reader.py
@@ -351,7 +351,7 @@ class Reader(object):
                if self._load_img:
                    sample['mixup']['image'] = self._load_image(sample['mixup'][
                        'im_file'])
-            if is_valid_path:
+            if is_valid_img_path:
                batch.append(sample)
                bs += 1
        return batch