From 7fce9632803f8a655e1870befa2892bb9bd179be Mon Sep 17 00:00:00 2001 From: Sangjin Date: Wed, 18 May 2022 16:14:14 +0900 Subject: [PATCH] fix(data): fix a bug occurs when mosaic_prob != 1 --- yolox/data/datasets/mosaicdetection.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/yolox/data/datasets/mosaicdetection.py b/yolox/data/datasets/mosaicdetection.py index 708babed5..500cb87ff 100644 --- a/yolox/data/datasets/mosaicdetection.py +++ b/yolox/data/datasets/mosaicdetection.py @@ -151,13 +151,13 @@ def __getitem__(self, idx): # img_info and img_id are not used for training. # They are also hard to be specified on a mosaic image. # ----------------------------------------------------------------- - return mix_img, padded_labels, img_info, img_id + return mix_img, padded_labels, img_info, idx else: self._dataset._input_dim = self.input_dim img, label, img_info, img_id = self._dataset.pull_item(idx) img, label = self.preproc(img, label, self.input_dim) - return img, label, img_info, img_id + return img, label, img_info, idx def mixup(self, origin_img, origin_labels, input_dim): jit_factor = random.uniform(*self.mixup_scale)