提交 58676dd6 编写于 作者: K Kaipeng Deng 提交者: GitHub

fix use _load_image (#2608)

上级 5ad237b6
...@@ -99,7 +99,7 @@ class RoiDbSource(Dataset): ...@@ -99,7 +99,7 @@ class RoiDbSource(Dataset):
mix_idx = random.randint(1, self._samples - 1) mix_idx = random.randint(1, self._samples - 1)
mix_pos = (mix_idx + self._pos) % self._samples mix_pos = (mix_idx + self._pos) % self._samples
sample['mixup'] = copy.deepcopy(self._roidb[mix_pos]) sample['mixup'] = copy.deepcopy(self._roidb[mix_pos])
if self._load_image: if self._load_img:
sample['mixup']['image'] = \ sample['mixup']['image'] = \
self._load_image(sample['mixup']['im_file']) self._load_image(sample['mixup']['im_file'])
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册