From 58676dd64dadf4ec38bc0398b902a3a4e3b0305a Mon Sep 17 00:00:00 2001 From: Kaipeng Deng Date: Fri, 28 Jun 2019 20:50:57 +0800 Subject: [PATCH] fix use _load_image (#2608) --- ppdet/data/source/roidb_source.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppdet/data/source/roidb_source.py b/ppdet/data/source/roidb_source.py index 15c99594d..1d9ff5d3b 100644 --- a/ppdet/data/source/roidb_source.py +++ b/ppdet/data/source/roidb_source.py @@ -99,7 +99,7 @@ class RoiDbSource(Dataset): mix_idx = random.randint(1, self._samples - 1) mix_pos = (mix_idx + self._pos) % self._samples sample['mixup'] = copy.deepcopy(self._roidb[mix_pos]) - if self._load_image: + if self._load_img: sample['mixup']['image'] = \ self._load_image(sample['mixup']['im_file']) else: -- GitLab