fix bug of normalize

db6c8253 · jiangjiajun · e7283a12 · db6c8253
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddlex/cv/transforms/ops.py paddlex/cv/transforms/ops.py +1 -1

未找到文件。
--- a/paddlex/cv/transforms/ops.py
+++ b/paddlex/cv/transforms/ops.py
@@ -26,7 +26,7 @@ def normalize(im, mean, std, min_value=[0, 0, 0], max_value=[255, 255, 255]):
    # Standardization (Z-score Normalization)
    im -= mean
    im /= std
-    return im
+    return im.astype('float32')


 def permute(im, to_bgr=False):