Browse Source

[fix] scale is not utilized in classification models (#1934)

zhang-prog 1 year ago
parent
commit
cb8f3b4513
1 changed files with 1 additions and 1 deletions
  1. 1 1
      paddlex/inference/models/image_classification.py

+ 1 - 1
paddlex/inference/models/image_classification.py

@@ -82,7 +82,7 @@ class ClasPredictor(CVPredictor):
     ):
         assert channel_num == 3
         assert order == ""
-        return Normalize(mean=mean, std=std)
+        return Normalize(scale=scale, mean=mean, std=std)
 
     @register("ToCHWImage")
     def build_to_chw(self):