Support https://huggingface.co/Revai/reverb-diarization-v1 (#1437)

2024-10-17 11:58:14 +08:00
parent 471cbd83c6
commit 620597f501
9 changed files with 276 additions and 12 deletions
--- a/scripts/pyannote/segmentation/vad-onnx.py
+++ b/scripts/pyannote/segmentation/vad-onnx.py
@@ -216,6 +216,8 @@ def main():

    is_active = classification[0] > onset
    start = None
+    if is_active:
+        start = 0

    scale = m.receptive_field_shift / m.sample_rate
    scale_offset = m.receptive_field_size / m.sample_rate * 0.5