소스 검색

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

gaotingquan 6 달 전
부모
커밋
e8d7f18a07
1개의 변경된 파일3개의 추가작업 그리고 4개의 파일을 삭제
  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):
         """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
 
     def __call__(self, preds):
-        normalized_features = [self._normalize(feature) for feature in preds]
-        return normalized_features
+        return self._normalize(preds[0])