EngineX-Iluvatar/enginex_bi_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Add C++ runtime for speaker verification models from NeMo (#527)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-01-13 21:42:09 +08:00
committed by GitHub
parent 68a525a024
commit 2024e96639
20 changed files with 405 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
sherpa-onnx/csrc/speaker-embedding-extractor-impl.cc
View File

@@ -1,11 +1,12 @@
// sherpa-onnx/csrc/speaker-embedding-extractor-impl.cc
//
// Copyright (c) 2023 Xiaomi Corporation
// Copyright (c) 2024 Xiaomi Corporation
#include "sherpa-onnx/csrc/speaker-embedding-extractor-impl.h"
#include "sherpa-onnx/csrc/macros.h"
#include "sherpa-onnx/csrc/onnx-utils.h"
#include "sherpa-onnx/csrc/speaker-embedding-extractor-general-impl.h"
#include "sherpa-onnx/csrc/speaker-embedding-extractor-nemo-impl.h"
namespace sherpa_onnx {
@@ -14,6 +15,7 @@ namespace {
enum class ModelType {
kWeSpeaker,
k3dSpeaker,
kNeMo,
kUnkown,
};
@@ -52,6 +54,8 @@ static ModelType GetModelType(char *model_data, size_t model_data_length,
return ModelType::kWeSpeaker;
} else if (model_type.get() == std::string("3d-speaker")) {
return ModelType::k3dSpeaker;
} else if (model_type.get() == std::string("nemo")) {
return ModelType::kNeMo;
} else {
SHERPA_ONNX_LOGE("Unsupported model_type: %s", model_type.get());
return ModelType::kUnkown;
@@ -74,6 +78,8 @@ SpeakerEmbeddingExtractorImpl::Create(
// fall through
case ModelType::k3dSpeaker:
return std::make_unique<SpeakerEmbeddingExtractorGeneralImpl>(config);
case ModelType::kNeMo:
return std::make_unique<SpeakerEmbeddingExtractorNeMoImpl>(config);
case ModelType::kUnkown:
SHERPA_ONNX_LOGE(
"Unknown model type in for speaker embedding extractor!");