Update data_aug.py (#40)

fix hsv_color_jitter bug

Update data_aug.py (#40)
fix hsv_color_jitter bug
e41f07ce · pennypm · Zeyu Chen · 2d04181b · e41f07ce
隐藏空白更改
内联并排

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
@@ -361,7 +361,7 @@ def hsv_color_jitter(crop_img,
    if brightness_jitter_ratio > 0 or \
        saturation_jitter_ratio > 0 or \
        contrast_jitter_ratio > 0:
-        random_jitter(crop_img, saturation_jitter_ratio,
+        crop_img = random_jitter(crop_img, saturation_jitter_ratio,
                      brightness_jitter_ratio, contrast_jitter_ratio)
    return crop_img