Browse Source

bugfix: error in feature calculating when bs > 1 in image feature model

gaotingquan 6 months ago
parent
commit
e8d7f18a07
1 changed files with 3 additions and 4 deletions
  1. 3 4
      paddlex/inference/models/image_feature/processors.py

+ 3 - 4
paddlex/inference/models/image_feature/processors.py

@@ -23,10 +23,9 @@ class NormalizeFeatures:
 
 
     def _normalize(self, preds):
     def _normalize(self, preds):
         """normalize"""
         """normalize"""
-        feas_norm = np.sqrt(np.sum(np.square(preds[0]), axis=0, keepdims=True))
-        features = np.divide(preds[0], feas_norm)
+        feas_norm = np.sqrt(np.sum(np.square(preds), axis=1, keepdims=True))
+        features = np.divide(preds, feas_norm)
         return features
         return features
 
 
     def __call__(self, preds):
     def __call__(self, preds):
-        normalized_features = [self._normalize(feature) for feature in preds]
-        return normalized_features
+        return self._normalize(preds[0])