From 0b48d1cc323a19a28cdcd43aec0f00190b5ab6b1 Mon Sep 17 00:00:00 2001 From: qipengfei7 Date: Sun, 26 Apr 2020 15:47:01 +0800 Subject: [PATCH] new robust reader --- ppdet/data/reader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppdet/data/reader.py b/ppdet/data/reader.py index f63e916c0..6d366fcae 100644 --- 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 -- GitLab