From 5845f8f25c61f56d39f67b78a6d0c25fe529eb25 Mon Sep 17 00:00:00 2001
From: LutaoChu <30695251+LutaoChu@users.noreply.github.com>
Date: Fri, 13 Dec 2019 10:45:25 +0800
Subject: [PATCH] fix color jitter bug (#119)

---
 pdseg/data_aug.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pdseg/data_aug.py b/pdseg/data_aug.py
index 15186150..933dc1b4 100644
--- a/pdseg/data_aug.py
+++ b/pdseg/data_aug.py
@@ -327,7 +327,7 @@ def random_jitter(cv_img, saturation_range, brightness_range, contrast_range):
     brightness_ratio = np.random.uniform(-brightness_range, brightness_range)
     contrast_ratio = np.random.uniform(-contrast_range, contrast_range)
 
-    order = [1, 2, 3]
+    order = [0, 1, 2]
     np.random.shuffle(order)
 
     for i in range(3):
-- 
GitLab