csukuangfj commited on
Commit
ba256cd
1 Parent(s): 8d68048

add revai sepaker segmentation models

Browse files
Files changed (1) hide show
  1. generate-speaker-diarization.py +8 -1
generate-speaker-diarization.py CHANGED
@@ -15,6 +15,7 @@ class APK:
15
  minor: int
16
  patch: int
17
  arch: str
 
18
 
19
  def __init__(self, s):
20
  # sherpa-onnx-1.10.27-arm64-v8a-speaker-diarization-pyannote_audio-3dspeaker.apk
@@ -22,6 +23,7 @@ class APK:
22
  split = s.split("-")
23
  self.major, self.minor, self.patch = list(map(int, split[2].split(".")))
24
  self.arch = split[3]
 
25
  if "arm" in s:
26
  self.arch += "-" + split[4]
27
 
@@ -34,7 +36,7 @@ class APK:
34
 
35
  def sort_by_apk(x):
36
  x = APK(x)
37
- return (x.major, x.minor, x.patch, x.arch)
38
 
39
 
40
  def get_all_files(d_list: List[str], suffix: str) -> List[str]:
@@ -123,6 +125,11 @@ at <a href="https://github.com/k2-fsa/sherpa-onnx/blob/master/scripts/apk/build-
123
  <td class="tg-0pky"><a href="https://github.com/k2-fsa/sherpa-onnx/releases/download/speaker-segmentation-models/sherpa-onnx-pyannote-segmentation-3-0.tar.bz2">sherpa-onnx-pyannote-segmentation-3-0.tar.bz2</a> It is <a href="https://github.com/k2-fsa/sherpa-onnx/tree/master/scripts/pyannote/segmentation">converted</a> from <a href="https://huggingface.co/pyannote/segmentation-3.0">https://huggingface.co/pyannote/segmentation-3.0</a></td>
124
  <td class="tg-0pky"><a href="https://github.com/k2-fsa/sherpa-onnx/releases/download/speaker-recongition-models/3dspeaker_speech_eres2net_base_sv_zh-cn_3dspeaker_16k.onnx">3dspeaker_speech_eres2net_base_sv_zh-cn_3dspeaker_16k.onnx</a> It is <a href="https://github.com/k2-fsa/sherpa-onnx/tree/master/scripts/3dspeaker">converted</a> from <a href="https://github.com/alibaba-damo-academy/3D-Speaker">https://github.com/alibaba-damo-academy/3D-Speaker</a></td>
125
  <tr>
 
 
 
 
 
126
  </tbody>
127
  </table>
128
 
 
15
  minor: int
16
  patch: int
17
  arch: str
18
+ name: str
19
 
20
  def __init__(self, s):
21
  # sherpa-onnx-1.10.27-arm64-v8a-speaker-diarization-pyannote_audio-3dspeaker.apk
 
23
  split = s.split("-")
24
  self.major, self.minor, self.patch = list(map(int, split[2].split(".")))
25
  self.arch = split[3]
26
+ self.name = split[-2]
27
  if "arm" in s:
28
  self.arch += "-" + split[4]
29
 
 
36
 
37
  def sort_by_apk(x):
38
  x = APK(x)
39
+ return (x.major, x.minor, x.patch, x.arch, self.name)
40
 
41
 
42
  def get_all_files(d_list: List[str], suffix: str) -> List[str]:
 
125
  <td class="tg-0pky"><a href="https://github.com/k2-fsa/sherpa-onnx/releases/download/speaker-segmentation-models/sherpa-onnx-pyannote-segmentation-3-0.tar.bz2">sherpa-onnx-pyannote-segmentation-3-0.tar.bz2</a> It is <a href="https://github.com/k2-fsa/sherpa-onnx/tree/master/scripts/pyannote/segmentation">converted</a> from <a href="https://huggingface.co/pyannote/segmentation-3.0">https://huggingface.co/pyannote/segmentation-3.0</a></td>
126
  <td class="tg-0pky"><a href="https://github.com/k2-fsa/sherpa-onnx/releases/download/speaker-recongition-models/3dspeaker_speech_eres2net_base_sv_zh-cn_3dspeaker_16k.onnx">3dspeaker_speech_eres2net_base_sv_zh-cn_3dspeaker_16k.onnx</a> It is <a href="https://github.com/k2-fsa/sherpa-onnx/tree/master/scripts/3dspeaker">converted</a> from <a href="https://github.com/alibaba-damo-academy/3D-Speaker">https://github.com/alibaba-damo-academy/3D-Speaker</a></td>
127
  <tr>
128
+ <tr>
129
+ <td class="tg-0pky">sherpa-onnx-x.y.z-arm64-v8a-speaker-diarization-revai_v1-3dspeaker.apk</td>
130
+ <td class="tg-0pky"><a href="https://github.com/k2-fsa/sherpa-onnx/releases/download/speaker-segmentation-models/sherpa-onnx-reverb-diarization-v1.tar.bz2">sherpa-onnx-reverb-diarization-v1.tar.bz2</a> It is <a href="https://github.com/k2-fsa/sherpa-onnx/tree/master/scripts/pyannote/segmentation">converted</a> from <a href="https://huggingface.co/Revai/reverb-diarization-v1">https://huggingface.co/Revai/reverb-diarization-v1</a></td>
131
+ <td class="tg-0pky"><a href="https://github.com/k2-fsa/sherpa-onnx/releases/download/speaker-recongition-models/3dspeaker_speech_eres2net_base_sv_zh-cn_3dspeaker_16k.onnx">3dspeaker_speech_eres2net_base_sv_zh-cn_3dspeaker_16k.onnx</a> It is <a href="https://github.com/k2-fsa/sherpa-onnx/tree/master/scripts/3dspeaker">converted</a> from <a href="https://github.com/alibaba-damo-academy/3D-Speaker">https://github.com/alibaba-damo-academy/3D-Speaker</a></td>
132
+ <tr>
133
  </tbody>
134
  </table>
135