diff --git a/pdseg/data_aug.py b/pdseg/data_aug.py index 15186150a3734a3a0c026386a04206ac036c7858..933dc1b442ab022b9d983ea8519b053ff3327c42 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):