diff --git a/ppdet/data/transform/operators.py b/ppdet/data/transform/operators.py index 8227b1db8e2489f296b299e9a609a2d251fc52e5..ba08c351e8b4f657dfe3f3173644ac1f785c7c38 100644 --- a/ppdet/data/transform/operators.py +++ b/ppdet/data/transform/operators.py @@ -2202,7 +2202,7 @@ class AugmentHSV(BaseOperator): else: cv2.cvtColor(img_hsv, cv2.COLOR_HSV2RGB, dst=img) - sample['image'] = img + sample['image'] = img.astype(np.float32) return sample