fix AugmentHSV output type (#5800)

ad0a436a · Guanghua Yu · GitHub · 96959b21 · ad0a436a
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppdet/data/transform/operators.py ppdet/data/transform/operators.py +1 -1

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