From ad0a436a12d45ec3cd13ae16bac79611908a872b Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Fri, 22 Apr 2022 11:43:04 +0800
Subject: [PATCH] fix AugmentHSV output type (#5800)

---
 ppdet/data/transform/operators.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppdet/data/transform/operators.py b/ppdet/data/transform/operators.py
index 8227b1db8..ba08c351e 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
 
 
-- 
GitLab