Drew commited on
Commit
9aee05d
Β·
1 Parent(s): 643a0dd

moved files

Browse files
This view is limited to 50 files because it contains too many changes. Β  See raw diff
Files changed (50) hide show
  1. segmentation/segmentation/LICENSE β†’ LICENSE +0 -0
  2. README.md +144 -3
  3. segmentation/segmentation/config.yaml β†’ config.yaml +0 -0
  4. segmentation/segmentation/example.png β†’ example.png +0 -0
  5. original.zip +3 -0
  6. segmentation/segmentation/pytorch_model.bin β†’ pytorch_model.bin +0 -0
  7. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/dihard3_custom_split/development.txt +0 -0
  8. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/dihard3_custom_split/train.txt +0 -0
  9. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/osd/AMI.development.rttm +0 -0
  10. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/osd/AMI.test.rttm +0 -0
  11. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/osd/DIHARD.development.rttm +0 -0
  12. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/osd/DIHARD.test.rttm +0 -0
  13. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/osd/VoxConverse.development.rttm +0 -0
  14. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/osd/VoxConverse.test.rttm +0 -0
  15. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/rsg/AMI.development.rttm +0 -0
  16. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/rsg/AMI.test.rttm +0 -0
  17. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/rsg/DIHARD.development.rttm +0 -0
  18. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/rsg/DIHARD.test.rttm +0 -0
  19. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/rsg/VoxConverse.development.rttm +0 -0
  20. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vad/AMI.development.rttm +0 -0
  21. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vad/AMI.test.rttm +0 -0
  22. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vad/DIHARD.development.rttm +0 -0
  23. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vad/DIHARD.test.rttm +0 -0
  24. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vad/VoxConverse.development.rttm +0 -0
  25. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vad/VoxConverse.test.rttm +0 -0
  26. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vbx/AMI.rttm +0 -0
  27. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vbx/DIHARD.rttm +0 -0
  28. {segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vbx/VoxConverse.rttm +0 -0
  29. segmentation/segmentation/.cache/huggingface/.gitignore +0 -1
  30. segmentation/segmentation/.cache/huggingface/download/.gitattributes.lock +0 -0
  31. segmentation/segmentation/.cache/huggingface/download/.gitattributes.metadata +0 -3
  32. segmentation/segmentation/.cache/huggingface/download/LICENSE.lock +0 -0
  33. segmentation/segmentation/.cache/huggingface/download/LICENSE.metadata +0 -3
  34. segmentation/segmentation/.cache/huggingface/download/README.md.lock +0 -0
  35. segmentation/segmentation/.cache/huggingface/download/README.md.metadata +0 -3
  36. segmentation/segmentation/.cache/huggingface/download/config.yaml.lock +0 -0
  37. segmentation/segmentation/.cache/huggingface/download/config.yaml.metadata +0 -3
  38. segmentation/segmentation/.cache/huggingface/download/example.png.lock +0 -0
  39. segmentation/segmentation/.cache/huggingface/download/example.png.metadata +0 -3
  40. segmentation/segmentation/.cache/huggingface/download/pytorch_model.bin.lock +0 -0
  41. segmentation/segmentation/.cache/huggingface/download/pytorch_model.bin.metadata +0 -3
  42. segmentation/segmentation/.cache/huggingface/download/reproducible_research/dihard3_custom_split/development.txt.lock +0 -0
  43. segmentation/segmentation/.cache/huggingface/download/reproducible_research/dihard3_custom_split/development.txt.metadata +0 -3
  44. segmentation/segmentation/.cache/huggingface/download/reproducible_research/dihard3_custom_split/train.txt.lock +0 -0
  45. segmentation/segmentation/.cache/huggingface/download/reproducible_research/dihard3_custom_split/train.txt.metadata +0 -3
  46. segmentation/segmentation/.cache/huggingface/download/reproducible_research/expected_outputs/osd/AMI.development.rttm.lock +0 -0
  47. segmentation/segmentation/.cache/huggingface/download/reproducible_research/expected_outputs/osd/AMI.development.rttm.metadata +0 -3
  48. segmentation/segmentation/.cache/huggingface/download/reproducible_research/expected_outputs/osd/AMI.test.rttm.lock +0 -0
  49. segmentation/segmentation/.cache/huggingface/download/reproducible_research/expected_outputs/osd/AMI.test.rttm.metadata +0 -3
  50. segmentation/segmentation/.cache/huggingface/download/reproducible_research/expected_outputs/osd/DIHARD.development.rttm.lock +0 -0
segmentation/segmentation/LICENSE β†’ LICENSE RENAMED
File without changes
README.md CHANGED
@@ -1,3 +1,144 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - pyannote
4
+ - pyannote-audio
5
+ - pyannote-audio-model
6
+ - audio
7
+ - voice
8
+ - speech
9
+ - speaker
10
+ - speaker-segmentation
11
+ - voice-activity-detection
12
+ - overlapped-speech-detection
13
+ - resegmentation
14
+ license: mit
15
+ inference: false
16
+ extra_gated_prompt: "The collected information will help acquire a better knowledge of pyannote.audio userbase and help its maintainers apply for grants to improve it further. If you are an academic researcher, please cite the relevant papers in your own publications using the model. If you work for a company, please consider contributing back to pyannote.audio development (e.g. through unrestricted gifts). We also provide scientific consulting services around speaker diarization and machine listening."
17
+ extra_gated_fields:
18
+ Company/university: text
19
+ Website: text
20
+ I plan to use this model for (task, type of audio data, etc): text
21
+ ---
22
+
23
+ Using this open-source model in production?
24
+ Consider switching to [pyannoteAI](https://www.pyannote.ai) for better and faster options.
25
+
26
+
27
+ # 🎹 Speaker segmentation
28
+
29
+ [Paper](http://arxiv.org/abs/2104.04045) | [Demo](https://huggingface.co/spaces/pyannote/pretrained-pipelines) | [Blog post](https://herve.niderb.fr/fastpages/2022/10/23/One-speaker-segmentation-model-to-rule-them-all)
30
+
31
+ ![Example](example.png)
32
+
33
+ ## Usage
34
+
35
+ Relies on pyannote.audio 2.1.1: see [installation instructions](https://github.com/pyannote/pyannote-audio).
36
+
37
+ ```python
38
+ # 1. visit hf.co/pyannote/segmentation and accept user conditions
39
+ # 2. visit hf.co/settings/tokens to create an access token
40
+ # 3. instantiate pretrained model
41
+ from pyannote.audio import Model
42
+ model = Model.from_pretrained("pyannote/segmentation",
43
+ use_auth_token="ACCESS_TOKEN_GOES_HERE")
44
+ ```
45
+
46
+ ### Voice activity detection
47
+
48
+ ```python
49
+ from pyannote.audio.pipelines import VoiceActivityDetection
50
+ pipeline = VoiceActivityDetection(segmentation=model)
51
+ HYPER_PARAMETERS = {
52
+ # onset/offset activation thresholds
53
+ "onset": 0.5, "offset": 0.5,
54
+ # remove speech regions shorter than that many seconds.
55
+ "min_duration_on": 0.0,
56
+ # fill non-speech regions shorter than that many seconds.
57
+ "min_duration_off": 0.0
58
+ }
59
+ pipeline.instantiate(HYPER_PARAMETERS)
60
+ vad = pipeline("audio.wav")
61
+ # `vad` is a pyannote.core.Annotation instance containing speech regions
62
+ ```
63
+
64
+ ### Overlapped speech detection
65
+
66
+ ```python
67
+ from pyannote.audio.pipelines import OverlappedSpeechDetection
68
+ pipeline = OverlappedSpeechDetection(segmentation=model)
69
+ pipeline.instantiate(HYPER_PARAMETERS)
70
+ osd = pipeline("audio.wav")
71
+ # `osd` is a pyannote.core.Annotation instance containing overlapped speech regions
72
+ ```
73
+
74
+ ### Resegmentation
75
+
76
+ ```python
77
+ from pyannote.audio.pipelines import Resegmentation
78
+ pipeline = Resegmentation(segmentation=model,
79
+ diarization="baseline")
80
+ pipeline.instantiate(HYPER_PARAMETERS)
81
+ resegmented_baseline = pipeline({"audio": "audio.wav", "baseline": baseline})
82
+ # where `baseline` should be provided as a pyannote.core.Annotation instance
83
+ ```
84
+
85
+ ### Raw scores
86
+
87
+ ```python
88
+ from pyannote.audio import Inference
89
+ inference = Inference(model)
90
+ segmentation = inference("audio.wav")
91
+ # `segmentation` is a pyannote.core.SlidingWindowFeature
92
+ # instance containing raw segmentation scores like the
93
+ # one pictured above (output)
94
+ ```
95
+
96
+
97
+ ## Citation
98
+
99
+ ```bibtex
100
+ @inproceedings{Bredin2021,
101
+ Title = {{End-to-end speaker segmentation for overlap-aware resegmentation}},
102
+ Author = {{Bredin}, Herv{\'e} and {Laurent}, Antoine},
103
+ Booktitle = {Proc. Interspeech 2021},
104
+ Address = {Brno, Czech Republic},
105
+ Month = {August},
106
+ Year = {2021},
107
+ ```
108
+
109
+ ```bibtex
110
+ @inproceedings{Bredin2020,
111
+ Title = {{pyannote.audio: neural building blocks for speaker diarization}},
112
+ Author = {{Bredin}, Herv{\'e} and {Yin}, Ruiqing and {Coria}, Juan Manuel and {Gelly}, Gregory and {Korshunov}, Pavel and {Lavechin}, Marvin and {Fustes}, Diego and {Titeux}, Hadrien and {Bouaziz}, Wassim and {Gill}, Marie-Philippe},
113
+ Booktitle = {ICASSP 2020, IEEE International Conference on Acoustics, Speech, and Signal Processing},
114
+ Address = {Barcelona, Spain},
115
+ Month = {May},
116
+ Year = {2020},
117
+ }
118
+ ```
119
+
120
+ ## Reproducible research
121
+
122
+ In order to reproduce the results of the paper ["End-to-end speaker segmentation for overlap-aware resegmentation
123
+ "](https://arxiv.org/abs/2104.04045), use `pyannote/segmentation@Interspeech2021` with the following hyper-parameters:
124
+
125
+ | Voice activity detection | `onset` | `offset` | `min_duration_on` | `min_duration_off` |
126
+ | ------------------------ | ------- | -------- | ----------------- | ------------------ |
127
+ | AMI Mix-Headset | 0.684 | 0.577 | 0.181 | 0.037 |
128
+ | DIHARD3 | 0.767 | 0.377 | 0.136 | 0.067 |
129
+ | VoxConverse | 0.767 | 0.713 | 0.182 | 0.501 |
130
+
131
+ | Overlapped speech detection | `onset` | `offset` | `min_duration_on` | `min_duration_off` |
132
+ | --------------------------- | ------- | -------- | ----------------- | ------------------ |
133
+ | AMI Mix-Headset | 0.448 | 0.362 | 0.116 | 0.187 |
134
+ | DIHARD3 | 0.430 | 0.320 | 0.091 | 0.144 |
135
+ | VoxConverse | 0.587 | 0.426 | 0.337 | 0.112 |
136
+
137
+ | Resegmentation of VBx | `onset` | `offset` | `min_duration_on` | `min_duration_off` |
138
+ | --------------------- | ------- | -------- | ----------------- | ------------------ |
139
+ | AMI Mix-Headset | 0.542 | 0.527 | 0.044 | 0.705 |
140
+ | DIHARD3 | 0.592 | 0.489 | 0.163 | 0.182 |
141
+ | VoxConverse | 0.537 | 0.724 | 0.410 | 0.563 |
142
+
143
+ Expected outputs (and VBx baseline) are also provided in the `/reproducible_research` sub-directories.
144
+
segmentation/segmentation/config.yaml β†’ config.yaml RENAMED
File without changes
segmentation/segmentation/example.png β†’ example.png RENAMED
File without changes
original.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b37ccf4b649e52e813551f4441e9dc9fb5f321f677f45fbba5cff77a80dd5167
3
+ size 57867411
segmentation/segmentation/pytorch_model.bin β†’ pytorch_model.bin RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/dihard3_custom_split/development.txt RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/dihard3_custom_split/train.txt RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/osd/AMI.development.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/osd/AMI.test.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/osd/DIHARD.development.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/osd/DIHARD.test.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/osd/VoxConverse.development.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/osd/VoxConverse.test.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/rsg/AMI.development.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/rsg/AMI.test.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/rsg/DIHARD.development.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/rsg/DIHARD.test.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/rsg/VoxConverse.development.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vad/AMI.development.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vad/AMI.test.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vad/DIHARD.development.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vad/DIHARD.test.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vad/VoxConverse.development.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vad/VoxConverse.test.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vbx/AMI.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vbx/DIHARD.rttm RENAMED
File without changes
{segmentation/segmentation/reproducible_research β†’ reproducible_research}/expected_outputs/vbx/VoxConverse.rttm RENAMED
File without changes
segmentation/segmentation/.cache/huggingface/.gitignore DELETED
@@ -1 +0,0 @@
1
- *
 
 
segmentation/segmentation/.cache/huggingface/download/.gitattributes.lock DELETED
File without changes
segmentation/segmentation/.cache/huggingface/download/.gitattributes.metadata DELETED
@@ -1,3 +0,0 @@
1
- 7cc8981665aa696b79c11c0339e73602af4c7350
2
- 12a25d60d1e877a9273c14f7336a9812664a06ab
3
- 1752243556.0054998
 
 
 
 
segmentation/segmentation/.cache/huggingface/download/LICENSE.lock DELETED
File without changes
segmentation/segmentation/.cache/huggingface/download/LICENSE.metadata DELETED
@@ -1,3 +0,0 @@
1
- 7cc8981665aa696b79c11c0339e73602af4c7350
2
- e5e0c2daded4524693e062d3e4fd016bbfb9a308
3
- 1752243555.817023
 
 
 
 
segmentation/segmentation/.cache/huggingface/download/README.md.lock DELETED
File without changes
segmentation/segmentation/.cache/huggingface/download/README.md.metadata DELETED
@@ -1,3 +0,0 @@
1
- 7cc8981665aa696b79c11c0339e73602af4c7350
2
- 5ff4ab94f198cf2882995ff4c1cb29e7db61e5c9
3
- 1752243555.994926
 
 
 
 
segmentation/segmentation/.cache/huggingface/download/config.yaml.lock DELETED
File without changes
segmentation/segmentation/.cache/huggingface/download/config.yaml.metadata DELETED
@@ -1,3 +0,0 @@
1
- 7cc8981665aa696b79c11c0339e73602af4c7350
2
- 3ad7b756881851b5ed1e58231cd3fac2b35bbfd8
3
- 1752243555.996554
 
 
 
 
segmentation/segmentation/.cache/huggingface/download/example.png.lock DELETED
File without changes
segmentation/segmentation/.cache/huggingface/download/example.png.metadata DELETED
@@ -1,3 +0,0 @@
1
- 7cc8981665aa696b79c11c0339e73602af4c7350
2
- be04026868f792563eccd82bd22c91719c828e71
3
- 1752243555.977073
 
 
 
 
segmentation/segmentation/.cache/huggingface/download/pytorch_model.bin.lock DELETED
File without changes
segmentation/segmentation/.cache/huggingface/download/pytorch_model.bin.metadata DELETED
@@ -1,3 +0,0 @@
1
- 7cc8981665aa696b79c11c0339e73602af4c7350
2
- 0b5b3216d60a2d32fc086b47ea8c67589aaeb26b7e07fcbe620d6d0b83e209ea
3
- 1752243556.188376
 
 
 
 
segmentation/segmentation/.cache/huggingface/download/reproducible_research/dihard3_custom_split/development.txt.lock DELETED
File without changes
segmentation/segmentation/.cache/huggingface/download/reproducible_research/dihard3_custom_split/development.txt.metadata DELETED
@@ -1,3 +0,0 @@
1
- 7cc8981665aa696b79c11c0339e73602af4c7350
2
- daaf8d50c108a254f1b4fc60ab1216977d9da274
3
- 1752243555.946626
 
 
 
 
segmentation/segmentation/.cache/huggingface/download/reproducible_research/dihard3_custom_split/train.txt.lock DELETED
File without changes
segmentation/segmentation/.cache/huggingface/download/reproducible_research/dihard3_custom_split/train.txt.metadata DELETED
@@ -1,3 +0,0 @@
1
- 7cc8981665aa696b79c11c0339e73602af4c7350
2
- 5949c6a6131ff986ef8f8e0cf70c6c94e4d696be
3
- 1752243555.810355
 
 
 
 
segmentation/segmentation/.cache/huggingface/download/reproducible_research/expected_outputs/osd/AMI.development.rttm.lock DELETED
File without changes
segmentation/segmentation/.cache/huggingface/download/reproducible_research/expected_outputs/osd/AMI.development.rttm.metadata DELETED
@@ -1,3 +0,0 @@
1
- 7cc8981665aa696b79c11c0339e73602af4c7350
2
- 3b135d9ac6e91a0891d4ed3e1339147fe7b7d1fa
3
- 1752243555.999325
 
 
 
 
segmentation/segmentation/.cache/huggingface/download/reproducible_research/expected_outputs/osd/AMI.test.rttm.lock DELETED
File without changes
segmentation/segmentation/.cache/huggingface/download/reproducible_research/expected_outputs/osd/AMI.test.rttm.metadata DELETED
@@ -1,3 +0,0 @@
1
- 7cc8981665aa696b79c11c0339e73602af4c7350
2
- da691e28da586cd4ff3be9ba22d0985ffc8548b9
3
- 1752243555.9985962
 
 
 
 
segmentation/segmentation/.cache/huggingface/download/reproducible_research/expected_outputs/osd/DIHARD.development.rttm.lock DELETED
File without changes