Add score function to speaker identification (#775)

k2-fsa · Apr 16, 2024 · aa2d695 · aa2d695
1 parent 6bf2099
commit aa2d695
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 0 deletions.
diff --git a/sherpa-onnx/csrc/speaker-embedding-manager.cc b/sherpa-onnx/csrc/speaker-embedding-manager.cc
@@ -151,6 +151,23 @@ class SpeakerEmbeddingManager::Impl {
     return true;
   }
 
+  float Score(const std::string &name, const float *p) {
+    if (!name2row_.count(name)) {
+      // Setting a default value if the name is not found
+      return -2.0;
+    }
+
+    int32_t row_idx = name2row_.at(name);
+
+    Eigen::VectorXf v =
+        Eigen::Map<Eigen::VectorXf>(const_cast<float *>(p), dim_);
+    v.normalize();
+
+    float score = embedding_matrix_.row(row_idx) * v;
+
+    return score;
+  }
+
   bool Contains(const std::string &name) const {
     return name2row_.count(name) > 0;
   }
@@ -206,6 +223,11 @@ bool SpeakerEmbeddingManager::Verify(const std::string &name, const float *p,
   return impl_->Verify(name, p, threshold);
 }
 
+float SpeakerEmbeddingManager::Score(const std::string &name,
+                                    const float *p) const {
+  return impl_->Score(name, p);
+}
+
 int32_t SpeakerEmbeddingManager::NumSpeakers() const {
   return impl_->NumSpeakers();
 }

diff --git a/sherpa-onnx/csrc/speaker-embedding-manager.h b/sherpa-onnx/csrc/speaker-embedding-manager.h
@@ -74,6 +74,8 @@ class SpeakerEmbeddingManager {
    */
   bool Verify(const std::string &name, const float *p, float threshold) const;
 
+  float Score(const std::string &name, const float *p) const;
+
   // Return true if the given speaker already exists; return false otherwise.
   bool Contains(const std::string &name) const;
 

diff --git a/sherpa-onnx/python/csrc/speaker-embedding-manager.cc b/sherpa-onnx/python/csrc/speaker-embedding-manager.cc
@@ -60,6 +60,14 @@ void PybindSpeakerEmbeddingManager(py::module *m) {
             return self.Verify(name, v.data(), threshold);
           },
           py::arg("name"), py::arg("v"), py::arg("threshold"),
+          py::call_guard<py::gil_scoped_release>())
+      .def(
+          "score",
+          [](const PyClass &self, const std::string &name,
+             const std::vector<float> &v) -> float {
+            return self.Score(name, v.data());
+          },
+          py::arg("name"), py::arg("v"),
           py::call_guard<py::gil_scoped_release>());
 }