diff --git a/data/hyps/hyp.scratch-high.yaml b/data/hyps/hyp.scratch-high.yaml index fdb2c3788..688efd755 100644 --- a/data/hyps/hyp.scratch-high.yaml +++ b/data/hyps/hyp.scratch-high.yaml @@ -28,3 +28,6 @@ fliplr: 0.5 # image flip left-right (probability) mosaic: 1.0 # image mosaic (probability) mixup: 0.15 # image mixup (probability) copy_paste: 0.3 # segment copy-paste (probability) +# assert bgr + gray + origin = 1 +bgr: 0.333 # bgr -> rgb (probability) +gray: 0.333 # brg -> gray (probability) diff --git a/utils/dataloaders.py b/utils/dataloaders.py index 776042999..64dadd110 100644 --- a/utils/dataloaders.py +++ b/utils/dataloaders.py @@ -706,6 +706,15 @@ def __getitem__(self, index): if nl: labels[:, 1] = 1 - labels[:, 1] + # gray and brg + _p = random.random() + if _p < hyp['bgr']: + img = img[..., ::-1] + elif _p < hyp['bgr'] + hyp['gray']: + img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY) + img = np.stack((img, img, img), axis=-1) + del _p + # Cutouts # labels = cutout(img, labels, p=0.5) # nl = len(labels) # update after cutout