Alkd commited on
Commit
edf8e8c
·
verified ·
1 Parent(s): db50522

Add model card

Browse files
Files changed (1) hide show
  1. README.md +29 -0
README.md ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-4.0
3
+ tags:
4
+ - onnx
5
+ - speaker-verification
6
+ - wespeaker
7
+ - pyannote
8
+ ---
9
+
10
+ # speaker-embedding-onnx
11
+
12
+ ONNX export of the ResNet34 backbone from [pyannote/wespeaker-voxceleb-resnet34-LM](https://huggingface.co/pyannote/wespeaker-voxceleb-resnet34-LM).
13
+
14
+ Follows the official [wespeaker/bin/export_onnx.py](https://github.com/wenet-e2e/wespeaker/blob/master/wespeaker/bin/export_onnx.py) approach: fbank features are computed externally, only the backbone is in ONNX.
15
+
16
+ ## Inputs / Outputs
17
+
18
+ | Name | Shape | Description |
19
+ |---|---|---|
20
+ | `input_features` | `(batch, T, 80)` | Kaldi fbank features (T is dynamic) |
21
+ | `embedding` | `(batch, 256)` | Speaker embedding vector |
22
+
23
+ ## Fbank parameters (must match at inference)
24
+
25
+ `kaldi.fbank(wav * 32768, num_mel_bins=80, frame_length=25, frame_shift=10,
26
+ round_to_power_of_two=True, window_type="hamming", use_energy=False,
27
+ snip_edges=True, dither=0.0, sample_frequency=16000)`
28
+
29
+ Then subtract per-bin mean: `feats -= feats.mean(axis=0)`.