Explorar o código

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

zhang-prog hai 1 ano
pai
achega
cb8f3b4513
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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):