Add Python API for clustering (#1385)

This commit is contained in:
Fangjun Kuang
2024-09-30 11:33:15 +08:00
committed by GitHub
parent 70568c2df7
commit b965f14cf0
26 changed files with 326 additions and 15 deletions

View File

@@ -35,6 +35,10 @@
#include "sherpa-onnx/python/csrc/offline-tts.h"
#endif
#if SHERPA_ONNX_ENABLE_SPEAKER_DIARIZATION == 1
#include "sherpa-onnx/python/csrc/fast-clustering.h"
#endif
namespace sherpa_onnx {
PYBIND11_MODULE(_sherpa_onnx, m) {
@@ -70,6 +74,10 @@ PYBIND11_MODULE(_sherpa_onnx, m) {
PybindOfflineTts(&m);
#endif
#if SHERPA_ONNX_ENABLE_SPEAKER_DIARIZATION == 1
PybindFastClustering(&m);
#endif
PybindSpeakerEmbeddingExtractor(&m);
PybindSpeakerEmbeddingManager(&m);
PybindSpokenLanguageIdentification(&m);