From 4bbfda5cdef05e8f841a8dfe03cf2c93d66a4f14 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 10 Apr 2020 17:29:57 -0700 Subject: [PATCH] hist equalization --- utils/datasets.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/utils/datasets.py b/utils/datasets.py index efd21f02..1755b177 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -521,9 +521,9 @@ def augment_hsv(img, hgain=0.5, sgain=0.5, vgain=0.5): 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]) + # if random.random() < 0.2: + # for i in range(3): + # img[:, :, i] = cv2.equalizeHist(img[:, :, i]) def load_mosaic(self, index):