Add score function to speaker identification (#775)

This commit is contained in:
chiiyeh
2024-04-16 17:29:46 +08:00
committed by GitHub
parent 6bf2099781
commit aa2d695fd2
3 changed files with 32 additions and 0 deletions

View File

@@ -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();
}

View File

@@ -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;