diff --git a/ppdet/data/source/roidb_source.py b/ppdet/data/source/roidb_source.py index 15c99594d7f586062125f7523e6e2e08e013f170..1d9ff5d3b39d4fb3ce17be5cbce2502e8d901a9a 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: