GGUF quantizations of the CED family (Consistent Ensemble Distillation, Xiaomi) - SOTA-tier audio-tagging models that classify everyday sounds (baby cry, footsteps, glass breaking, alarms, dog bark,...) into the 527-class AudioSet ontology. These files run with ced.cpp, a standalone C++/ggml port (no Python, no PyTorch at inference), and with LocalAI via the ced backend. Converted from the mispeech/ced- checkpoints (Apache-2.0). CED is a plain AST/DeiT Vision Transformer over a log-mel spectrogram; the port is numerically equal to the PyTorch reference. One self-contained GGUF per size + quant (config, 527 labels, and the mel filterbank/window are all embedded). Pick by your accuracy/size…
Open-weight model · Audio classification
pyannote-speaker-diarization.axera
by ZY HY-2012/pyannote-speaker-diarization.axera
pyannote-speaker-diarization.axera is an open-weight model for audio classification from ZY, released under Creative Commons Attribution 4.0. Its published files total 297.3 MB. It draws 35 downloads a month.
pyannote/speaker-diarization-community-1(cc-by-4.0)说话人日志模型的 AX650N 量化部署包: 输入 16 kHz 单声道会议音频,输出 RTTM 说话人时序标签。 - 模型: 8 个 axmodel(U16 激活 / S8 权重) - 依赖: Python 仅 numpy + scipy + scikit-learn + soundfile + axengine;C++ 为预编译 aarch64 可执行文件 常用参数:--step 2.0(滑动步长,2.5…
Model Card
By ZY, published under cc-by-4.0, revision 9d7333a0521b.
pyannote/speaker-diarization-community-1(cc-by-4.0)说话人日志模型的 AX650N 量化部署包: 输入 16 kHz 单声道会议音频,输出 RTTM 说话人时序标签。 - 模型: 8 个 axmodel(U16 激活 / S8 权重) - 依赖: Python 仅 numpy + scipy + scikit-learn + soundfile + axengine;C++ 为预编译 aarch64 可执行文件 常用参数:--step 2.0(滑动步长,2.5 更快 +0~1.5 pp)、--num-speakers 0(0=自动估计)。 切换 NPU2 版:--model-dir modelsnpu2(runax650.sh 通过环境变量 MODELDIR 指定)。 - 输入: 16 kHz 单声道 wav;输出: RTTM(SPEAKER 1 SPEAKERXX ) - 流水线: sincnet 分割(im2col + MatMul)→ 4 层 BiLSTM(96 步 cell,FP32 状态跨块) → 说话人嵌入(ResNet34 + TSTP)→ PLDA/AHC/VBx 聚类 - InstanceNorm / TSTP 统计池化在 U16 量化域数值崩坏,放主机 FP32 计算 - 完整转换源码见 GitHub: pyannote-speaker-diarization.AXERA 指标表所用 AMI / AliMeeting / VoxConverse 三个测试集的下载链接、参考处理与评分脚本见 ¹…
Read ZY's full model card
pyannote/speaker-diarization-community-1(cc-by-4.0)说话人日志模型的 AX650N 量化部署包:
输入 16 kHz 单声道会议音频,输出 RTTM 说话人时序标签。
- 模型: 8 个 axmodel(U16 激活 / S8 权重)
- 依赖: Python 仅 numpy + scipy + scikit-learn + soundfile + axengine;C++ 为预编译 aarch64 可执行文件
目录
├── models/ # 8 个 axmodel + 主机参数(NPU3 三核版,推荐)
├── models_npu2/ # 同一组模型 NPU2 双核版(输出与 NPU3 一致,RTF 慢约 33%)
├── python/ # Python 推理入口(community1_sdk + example.py)
├── bin/ # C++ 可执行文件(aarch64):community1_diar_ax650
├── samples/ # 演示音频(2 人会议 120 s)
├── run_ax650.sh run_cpp_ax650.sh # 一键运行(Python / C++)
└── requirements.txt
Python 运行
pip install numpy scipy scikit-learn soundfile
# pyaxengine: https://github.com/AXERA-TECH/pyaxengine/releases/latest
pip3 install axengine-<version>-py3-none-any.whl
./run_ax650.sh # 自带样例 samples/sample_meeting.wav
./run_ax650.sh input.wav output.rttm # 自定义输入
或:
from community1_sdk.pipeline import Pipeline, run_pipeline
pipeline = Pipeline(seg_a='models/segmentation_sincnet.axmodel', ...)
rttm = run_pipeline(pipeline, wav_16k_float32, uri='audio')
C++ 运行
export LD_LIBRARY_PATH=/soc/lib:${LD_LIBRARY_PATH:-}
./bin/community1_diar_ax650 --model-dir models --wav input.wav \
--out output.rttm --step 2.0
# 或一键:./run_cpp_ax650.sh input.wav output.rttm
常用参数:--step 2.0(滑动步长,2.5 更快 +0~1.5 pp)、--num-speakers 0(0=自动估计)。
切换 NPU2 版:--model-dir models_npu2(run_ax650.sh 通过环境变量 MODEL_DIR 指定)。
模型说明
- 输入: 16 kHz 单声道 wav;输出: RTTM(
SPEAKER <uri> 1 <start> <dur> <NA> <NA> SPEAKER_XX <NA> <NA>) - 流水线: sincnet 分割(im2col + MatMul)→ 4 层 BiLSTM(96 步 cell,FP32 状态跨块) → 说话人嵌入(ResNet34 + TSTP)→ PLDA/AHC/VBx 聚类
- InstanceNorm / TSTP 统计池化在 U16 量化域数值崩坏,放主机 FP32 计算
- 完整转换源码见 GitHub: pyannote-speaker-diarization.AXERA
测试集评测
指标表所用 AMI / AliMeeting / VoxConverse 三个测试集的下载链接、参考处理与评分脚本见 GitHub benchmark/。
RTF(AX650N 实测, samples/sample_meeting.wav 120 s)
| 推理路径 | 耗时 | RTF |
|---|---|---|
| C++(8 线程, step=2.0) | 6.9 s | 0.057 |
| Python(step=2.0) | 19 s | 0.16 |
RTF = 推理耗时 / 音频时长(不含模型加载);单场 30 分钟会议约 2.5 分钟(C++)。
指标(帧级 DER)
| 数据集 | community-1 板端 | 3D-Speaker 板端¹ | community-1 GPU |
|---|---|---|---|
| AMI dev12(no collar) | 20.06% | 29.72% | 20.08% |
| AliMeeting eval(±0.125 s) | 20.88% | 29.34% | 18.84% |
| AliMeeting eval(±0.25 s) | 17.26% | 24.36% | 15.16% |
| VoxConverse test(±0.125 s) | 9.11% | 9.27% | 8.49% |
¹ 3D-Speaker 板端 = FSMN VAD + CAM++ + 谱聚类量化管线,RTF 0.046~0.047。
参考
- pyannote-audio(MIT)
- pyannote-speaker-diarization.AXERA(完整转换源码)
Identity and Version
- Repository
- HY-2012/pyannote-speaker-diarization.axera
- Publisher
- ZY
- Task
- Audio classification
- Modality
- Audio
- Library
- Not stated by the source
- Parameters
- Not stated by the source
- Languages
- zh, en
- Revision
- 9d7333a0521b4d4c6971838f4965e87fdaf344b2
- First published
- 2026-09-08
- Last updated
- 2026-10-09
Files and Weights
63 files, 297.3 MB in total. The weights are 28 files totalling 34.1 MB in bin, npz.
Every file
| File | Type | Size | SHA-256 |
|---|---|---|---|
| models/embtail.bin | Weights | 5.2 MB | c2a1586fa656 |
| models/fbank.bin | Weights | 83.9 KB | de2fbe0c2b7d |
| models/fbank_params.npz | Weights | 85.4 KB | 0fa82ca61a79 |
| models/ffn.bin | Weights | 201.2 KB | d6f624b132dd |
| models/ffn_params.npz | Weights | 202.7 KB | 1ee1a8180e38 |
| models/frontend.bin | Weights | 1.6 KB | da0529aeb15c |
| models/host_emb_tail.npz | Weights | 5.2 MB | e8c1e8da0423 |
| models/host_frontend.npz | Weights | 4.6 KB | d282e67227f6 |
| models/lstm_params.npz | Weights | 5.5 MB | 91d515f1c8d7 |
| models/plda.bin | Weights | 199.2 KB | 5b50da1827c4 |
| models/plda.npz | Weights | 133.9 KB | 9b77bcd84069 |
| models/powerset.bin | Weights | 84 B | 1feb393d1e1f |
| models/powerset_mapping.npz | Weights | 352 B | 0a52d770a47f |
| models/xvec_transform.npz | Weights | 134.4 KB | 325f1ce8e48f |
| models_npu2/embtail.bin | Weights | 5.2 MB | c2a1586fa656 |
| models_npu2/fbank.bin | Weights | 83.9 KB | de2fbe0c2b7d |
| models_npu2/fbank_params.npz | Weights | 85.4 KB | 0fa82ca61a79 |
| models_npu2/ffn.bin | Weights | 201.2 KB | d6f624b132dd |
| models_npu2/ffn_params.npz | Weights | 202.7 KB | 1ee1a8180e38 |
| models_npu2/frontend.bin | Weights | 1.6 KB | da0529aeb15c |
| models_npu2/host_emb_tail.npz | Weights | 5.2 MB | e8c1e8da0423 |
| models_npu2/host_frontend.npz | Weights | 4.6 KB | d282e67227f6 |
| models_npu2/lstm_params.npz | Weights | 5.5 MB | 91d515f1c8d7 |
| models_npu2/plda.bin | Weights | 199.2 KB | 5b50da1827c4 |
| models_npu2/plda.npz | Weights | 133.9 KB | 9b77bcd84069 |
| models_npu2/powerset.bin | Weights | 84 B | 1feb393d1e1f |
| models_npu2/powerset_mapping.npz | Weights | 352 B | 0a52d770a47f |
| models_npu2/xvec_transform.npz | Weights | 134.4 KB | 325f1ce8e48f |
| config.json | Configuration | 3 B | — |
| configuration.json | Configuration | 256 B | — |
| models/model_meta.json | Configuration | 2.1 KB | — |
| models_npu2/model_meta.json | Configuration | 2.1 KB | — |
| python/community1_sdk/__init__.py | Configuration | 186 B | — |
| python/community1_sdk/axwrap.py | Configuration | 1.7 KB | — |
| python/community1_sdk/fbank.py | Configuration | 3.2 KB | — |
| python/community1_sdk/ffn.py | Configuration | 1.4 KB | — |
| python/community1_sdk/lstm.py | Configuration | 4.7 KB | — |
| python/community1_sdk/pipeline.py | Configuration | 13.8 KB | — |
| python/community1_sdk/vbx.py | Configuration | 3.6 KB | — |
| python/example.py | Configuration | 3.9 KB | — |
| README.md | Documentation | 3.8 KB | — |
| bin/community1_diar_ax650 | Other | 136.2 KB | e0560852c461 |
| models/embedding_convs.axmodel | Other | 5.6 MB | 711d572e2752 |
| models/lstm_cell_l0.axmodel | Other | 30.6 MB | a4260c6faf90 |
| models/lstm_cell_l1.axmodel | Other | 31.1 MB | 4b537cb75cfa |
| models/lstm_cell_l2.axmodel | Other | 31.0 MB | 6f0fa7ac0d1c |
| models/lstm_cell_l3.axmodel | Other | 30.9 MB | 6a00fd895889 |
| models/segmentation_convs1.axmodel | Other | 43.0 KB | — |
| models/segmentation_convs2.axmodel | Other | 34.8 KB | — |
| models/segmentation_sincnet.axmodel | Other | 50.0 KB | — |
| models_npu2/embedding_convs.axmodel | Other | 5.6 MB | bc64bdbb5faf |
| models_npu2/lstm_cell_l0.axmodel | Other | 30.6 MB | 44a6a2c2ead7 |
| models_npu2/lstm_cell_l1.axmodel | Other | 31.1 MB | ce8553c561fd |
| models_npu2/lstm_cell_l2.axmodel | Other | 31.0 MB | 8f959806337f |
| models_npu2/lstm_cell_l3.axmodel | Other | 30.9 MB | dbe3594fbfbd |
| models_npu2/segmentation_convs1.axmodel | Other | 39.0 KB | — |
| models_npu2/segmentation_convs2.axmodel | Other | 31.3 KB | — |
| models_npu2/segmentation_sincnet.axmodel | Other | 438.0 KB | db171fb319c7 |
| requirements.txt | Other | 44 B | — |
| run_ax650.sh | Other | 1.0 KB | — |
| run_cpp_ax650.sh | Other | 514 B | — |
| samples/sample_meeting.wav | Other | 3.8 MB | 81bd9e5314f1 |
| .gitattributes | Repository | 2.8 KB | — |
License and Download
- License
- cc-by-4.0
- Access
- Open weights, no gate
- Download size
- 34.1 MB
Released by ZY through its official repository on Hugging Face. Read the license.
Memory Requirements
| Precision | Weights in memory |
|---|---|
| As published | 34.1 MB |
Weights only, from the published parameter count; the key-value cache and runtime add to this.
Questions About pyannote-speaker-diarization.axera
Can I use pyannote-speaker-diarization.axera commercially?
Yes. pyannote-speaker-diarization.axera is released under Creative Commons Attribution 4.0. CC BY 4.0 permits sharing and adapting the work, including commercially, provided the creator is credited and changes are indicated.
Similar Models
This is a spoken language recognition model trained on the VoxLingua107 dataset using SpeechBrain. The model uses the ECAPA-TDNN architecture that has previously been used for speaker recognition. However, it uses more fully connected hidden layers after the embedding layer, and cross-entropy loss was used for training. We observed that this improved the performance of extracted utterance embeddings for downstream tasks. The system is trained with recordings sampled at 16kHz (single channel). The code will automatically normalize your audio (i.e., resampling + mono channel selection) when calling classifyfile if needed. The model can classify a speech utterance according to the language…
Abstract: The recognition of accented speech still remains a dominant problem in Automatic Speech Recognition (ASR) systems. We approach the classification of accented English speech through the Emphasized Channel Attention, Propagation and Aggregation Time Delay Neural Network (ECAPA-TDNN) architecture which has been shown to perform well on a variety of speech tasks. Three models are proposed: one trained from scratch, another two models (one using data augmentation and a baseline model) fine-tuned from the checkpoints of speechbrain/spkrec-ecapa-voxceleb (VoxCeleb). Our results show that the model fine-tuned with data augmentation yield the best results. Most of the misclassifications…
The development log of our Music Audio Pre-training (m-a-p) model family: - 02/06/2023: arxiv pre-print and training codes released. - 17/03/2023: we release two advanced music understanding models, MERT-v1-95M and MERT-v1-330M, trained with new paradigm and dataset. They outperform the previous models and can better generalize to more tasks. - 14/03/2023: we retrained the MERT-v0 model with open-source-only music dataset MERT-v0-public - 29/12/2022: a music understanding model MERT-v0 trained with MLM paradigm, which performs better at downstream tasks. - 29/10/2022: a pre-trained MIR model music2vec trained with BYOL paradigm. Here is a table for quick model pick-up: The m-a-p models…
The development log of our Music Audio Pre-training (m-a-p) model family: - 02/06/2023: arxiv pre-print and training codes released. - 17/03/2023: we release two advanced music understanding models, MERT-v1-95M and MERT-v1-330M, trained with new paradigm and dataset. They outperform the previous models and can better generalize to more tasks. - 14/03/2023: we retrained the MERT-v0 model with open-source-only music dataset MERT-v0-public - 29/12/2022: a music understanding model MERT-v0 trained with MLM paradigm, which performs better at downstream tasks. - 29/10/2022: a pre-trained MIR model music2vec trained with BYOL paradigm. Here is a table for quick model pick-up: The m-a-p models…
This model combines the SpeechBrain ECAPA-TDNN speaker embedding model with an SVM classifier to predict speaker gender from audio input. The model was trained and evaluated on the VoxCeleb2, Mozilla Common Voice v10.0, and TIMIT datasets - Mozilla Common Voice v10.0 English validated test set: 92.3% accuracy The model was trained on VoxCeleb2 dataset: - Converted to WAV format, single channel, 16kHz sampling rate, 256 kp/s bitrate - Applied SileroVAD for voice activity detection, taking the first voiced segment You can install the package directly from GitHub: - Model was trained on celebrity voices from YouTube interviews - Performance may vary on different audio qualities or recording…
