bug fix: audio feature extract and index mistake when register new sp…

…eaker
lovemefan · Dec 16, 2023 · 11738c0 · 11738c0
1 parent b9202eb
commit 11738c0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/paraformer/runtime/python/model/sv/campplus.py b/paraformer/runtime/python/model/sv/campplus.py
@@ -82,7 +82,7 @@ def extract_feature(self, audio: Union[str, Path, bytes], sample_rate=16000):
             mat[i, :] = fbank_fn.get_frame(i)
         feature = mat.astype(np.float32)
 
-        feature = feature - feature.mean()
+        feature = feature - feature.mean(0, keepdims=True)
         feature = feature[None, ...]
         return feature
 
@@ -109,5 +109,6 @@ def recognize(self, waveform: Union[str, Path, bytes], threshold=0.65):
 
         if sim[max_sim_index] <= threshold:
             self.register_speaker(emb)
+            max_sim_index = len(self.memory) - 1
 
         return max_sim_index