csukuangfj commited on
Commit
0069ad3
1 Parent(s): e2607c2

small fixes

Browse files
Files changed (1) hide show
  1. generate-speaker-diarization.py +19 -7
generate-speaker-diarization.py CHANGED
@@ -15,7 +15,8 @@ class APK:
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,7 +24,8 @@ class 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,7 +38,7 @@ class APK:
36
 
37
  def sort_by_apk(x):
38
  x = APK(x)
39
- return (x.major, x.minor, x.patch, x.arch, x.name)
40
 
41
 
42
  def get_all_files(d_list: List[str], suffix: str) -> List[str]:
@@ -122,14 +124,24 @@ at <a href="https://github.com/k2-fsa/sherpa-onnx/blob/master/scripts/apk/build-
122
  <tbody>
123
  <tr>
124
  <td class="tg-0pky">sherpa-onnx-x.y.z-arm64-v8a-speaker-diarization-pyannote_audio-3dspeaker.apk</td>
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
 
 
15
  minor: int
16
  patch: int
17
  arch: str
18
+ name1: str
19
+ name2: str
20
 
21
  def __init__(self, s):
22
  # sherpa-onnx-1.10.27-arm64-v8a-speaker-diarization-pyannote_audio-3dspeaker.apk
 
24
  split = s.split("-")
25
  self.major, self.minor, self.patch = list(map(int, split[2].split(".")))
26
  self.arch = split[3]
27
+ self.name1 = split[-2]
28
+ self.name2 = split[-1]
29
  if "arm" in s:
30
  self.arch += "-" + split[4]
31
 
 
38
 
39
  def sort_by_apk(x):
40
  x = APK(x)
41
+ return (x.major, x.minor, x.patch, x.arch, x.name1, x.name2)
42
 
43
 
44
  def get_all_files(d_list: List[str], suffix: str) -> List[str]:
 
124
  <tbody>
125
  <tr>
126
  <td class="tg-0pky">sherpa-onnx-x.y.z-arm64-v8a-speaker-diarization-pyannote_audio-3dspeaker.apk</td>
127
+ <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> <br/><br/>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>
128
+ <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> <br/><br/>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>
129
+ </tr>
130
  <tr>
131
+ <td class="tg-0pky">sherpa-onnx-x.y.z-arm64-v8a-speaker-diarization-pyannote_audio-nemo.apk</td>
132
+ <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> <br/><br/>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>
133
+ <td class="tg-0pky"><a href="https://github.com/k2-fsa/sherpa-onnx/releases/download/speaker-recongition-models/nemo_en_titanet_small.onnx">nemo_en_titanet_small.onnx</a> <br/><br/>It is <a href="https://github.com/k2-fsa/sherpa-onnx/tree/master/scripts/nemo/speaker-verification">converted</a> from <a href="https://catalog.ngc.nvidia.com/orgs/nvidia/teams/nemo/models/titanet_small">https://catalog.ngc.nvidia.com/orgs/nvidia/teams/nemo/models/titanet_small</a></td>
134
+ </tr>
135
  <tr>
136
  <td class="tg-0pky">sherpa-onnx-x.y.z-arm64-v8a-speaker-diarization-revai_v1-3dspeaker.apk</td>
137
+ <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> <br/><br/>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>
138
+ <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> <br/><br/>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>
139
+ </tr>
140
  <tr>
141
+ <td class="tg-0pky">sherpa-onnx-x.y.z-arm64-v8a-speaker-diarization-revai_v1-nemo.apk</td>
142
+ <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> <br/><br/>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>
143
+ <td class="tg-0pky"><a href="https://github.com/k2-fsa/sherpa-onnx/releases/download/speaker-recongition-models/nemo_en_titanet_small.onnx">nemo_en_titanet_small.onnx</a><br/><br/> It is <a href="https://github.com/k2-fsa/sherpa-onnx/tree/master/scripts/nemo/speaker-verification">converted</a> from <a href="https://catalog.ngc.nvidia.com/orgs/nvidia/teams/nemo/models/titanet_small">https://catalog.ngc.nvidia.com/orgs/nvidia/teams/nemo/models/titanet_small</a></td>
144
+ </tr>
145
  </tbody>
146
  </table>
147