fix color jitter bug (#119)

5845f8f2 · LutaoChu · wuzewu · d47b726c · 5845f8f2
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

pdseg/data_aug.py pdseg/data_aug.py +1 -1

未找到文件。
--- 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):