diff --git a/utils/datasets.py b/utils/datasets.py index 8f1341e3..efd21f02 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -520,6 +520,11 @@ def augment_hsv(img, hgain=0.5, sgain=0.5, vgain=0.5): np.clip(img_hsv[:, :, 0], None, 179, out=img_hsv[:, :, 0]) # inplace hue clip (0 - 179 deg) cv2.cvtColor(img_hsv, cv2.COLOR_HSV2BGR, dst=img) # no return needed + # Histogram equalization + if random.random() < 0.2: + for i in range(3): + img[:, :, i] = cv2.equalizeHist(img[:, :, i]) + def load_mosaic(self, index): # loads images in a mosaic