Python API for speaker diarization. (#1400)

2024-10-09 14:13:26 +08:00
parent 59407edcad
commit 8535b1d3bb
14 changed files with 315 additions and 9 deletions
--- a/sherpa-onnx/csrc/offline-speaker-diarization.h
+++ b/sherpa-onnx/csrc/offline-speaker-diarization.h
@@ -34,10 +34,13 @@ struct OfflineSpeakerDiarizationConfig {
  OfflineSpeakerDiarizationConfig(
      const OfflineSpeakerSegmentationModelConfig &segmentation,
      const SpeakerEmbeddingExtractorConfig &embedding,
-      const FastClusteringConfig &clustering)
+      const FastClusteringConfig &clustering, float min_duration_on,
+      float min_duration_off)
      : segmentation(segmentation),
        embedding(embedding),
-        clustering(clustering) {}
+        clustering(clustering),
+        min_duration_on(min_duration_on),
+        min_duration_off(min_duration_off) {}

  void Register(ParseOptions *po);
  bool Validate() const;