bbd7c7fc18
See pre-built Android APKs at https://k2-fsa.github.io/sherpa/onnx/speaker-identification/apk.html
67 lines
2.4 KiB
C++
67 lines
2.4 KiB
C++
// sherpa-onnx/python/csrc/speaker-embedding-manager.cc
|
|
//
|
|
// Copyright (c) 2023 Xiaomi Corporation
|
|
|
|
#include "sherpa-onnx/python/csrc/speaker-embedding-manager.h"
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "sherpa-onnx/csrc/speaker-embedding-manager.h"
|
|
|
|
namespace sherpa_onnx {
|
|
|
|
void PybindSpeakerEmbeddingManager(py::module *m) {
|
|
using PyClass = SpeakerEmbeddingManager;
|
|
py::class_<PyClass>(*m, "SpeakerEmbeddingManager")
|
|
.def(py::init<int32_t>(), py::arg("dim"),
|
|
py::call_guard<py::gil_scoped_release>())
|
|
.def_property_readonly("num_speakers", &PyClass::NumSpeakers)
|
|
.def_property_readonly("dim", &PyClass::Dim)
|
|
.def_property_readonly("all_speakers", &PyClass::GetAllSpeakers)
|
|
.def(
|
|
"__contains__",
|
|
[](const PyClass &self, const std::string &name) -> bool {
|
|
return self.Contains(name);
|
|
},
|
|
py::arg("name"), py::call_guard<py::gil_scoped_release>())
|
|
.def(
|
|
"add",
|
|
[](const PyClass &self, const std::string &name,
|
|
const std::vector<float> &v) -> bool {
|
|
return self.Add(name, v.data());
|
|
},
|
|
py::arg("name"), py::arg("v"),
|
|
py::call_guard<py::gil_scoped_release>())
|
|
.def(
|
|
"add",
|
|
[](const PyClass &self, const std::string &name,
|
|
const std::vector<std::vector<float>> &embedding_list) -> bool {
|
|
return self.Add(name, embedding_list);
|
|
},
|
|
py::arg("name"), py::arg("embedding_list"),
|
|
py::call_guard<py::gil_scoped_release>())
|
|
.def(
|
|
"remove",
|
|
[](const PyClass &self, const std::string &name) -> bool {
|
|
return self.Remove(name);
|
|
},
|
|
py::arg("name"), py::call_guard<py::gil_scoped_release>())
|
|
.def(
|
|
"search",
|
|
[](const PyClass &self, const std::vector<float> &v, float threshold)
|
|
-> std::string { return self.Search(v.data(), threshold); },
|
|
py::arg("v"), py::arg("threshold"),
|
|
py::call_guard<py::gil_scoped_release>())
|
|
.def(
|
|
"verify",
|
|
[](const PyClass &self, const std::string &name,
|
|
const std::vector<float> &v, float threshold) -> bool {
|
|
return self.Verify(name, v.data(), threshold);
|
|
},
|
|
py::arg("name"), py::arg("v"), py::arg("threshold"),
|
|
py::call_guard<py::gil_scoped_release>());
|
|
}
|
|
|
|
} // namespace sherpa_onnx
|