From 57305f40496d493dbac18a15b0ae993d8159fc6d Mon Sep 17 00:00:00 2001 From: jason_w Date: Fri, 30 Sep 2022 11:03:15 +0800 Subject: [PATCH 1/2] clip saturation value to valid range [0, 1] --- mmdet/datasets/transforms/transforms.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mmdet/datasets/transforms/transforms.py b/mmdet/datasets/transforms/transforms.py index e28ca9c2d2a..bbc229bb2a8 100644 --- a/mmdet/datasets/transforms/transforms.py +++ b/mmdet/datasets/transforms/transforms.py @@ -981,6 +981,10 @@ def transform(self, results: dict) -> dict: # random saturation if saturation_flag: img[..., 1] *= saturation_value + # For image(type=float32), after convert bgr to hsv by opencv, + # valid saturation value range is [0, 1] + if saturation_value > 1: + img[..., 1][img[..., 1] > 1] = 1 # random hue if hue_flag: From 229c96429e6b920e6df6562c56a8bf4421ad2f1b Mon Sep 17 00:00:00 2001 From: jason_w Date: Mon, 28 Nov 2022 15:24:40 +0800 Subject: [PATCH 2/2] Update mmdet/datasets/transforms/transforms.py Co-authored-by: RangiLyu --- mmdet/datasets/transforms/transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmdet/datasets/transforms/transforms.py b/mmdet/datasets/transforms/transforms.py index bbc229bb2a8..7ae88dcc568 100644 --- a/mmdet/datasets/transforms/transforms.py +++ b/mmdet/datasets/transforms/transforms.py @@ -984,7 +984,7 @@ def transform(self, results: dict) -> dict: # For image(type=float32), after convert bgr to hsv by opencv, # valid saturation value range is [0, 1] if saturation_value > 1: - img[..., 1][img[..., 1] > 1] = 1 + img[..., 1] = img[..., 1].clip(0, 1) # random hue if hue_flag: