Skip to page content
SAVRN
Search Plan a project

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…

Parameters—
Context—
Weights34.1 MB
Licensecc-by-4.0
AccessOpen weights
Monthly Downloads35

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。

参考

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.

Weights28 files · 34.1 MB
Configuration12 files · 36.8 KB
Documentation1 file · 3.8 KB
Other21 files · 263.2 MB
Repository1 file · 2.8 KB
Every file
FileTypeSizeSHA-256
models/embtail.binWeights5.2 MB c2a1586fa656
models/fbank.binWeights83.9 KB de2fbe0c2b7d
models/fbank_params.npzWeights85.4 KB 0fa82ca61a79
models/ffn.binWeights201.2 KB d6f624b132dd
models/ffn_params.npzWeights202.7 KB 1ee1a8180e38
models/frontend.binWeights1.6 KB da0529aeb15c
models/host_emb_tail.npzWeights5.2 MB e8c1e8da0423
models/host_frontend.npzWeights4.6 KB d282e67227f6
models/lstm_params.npzWeights5.5 MB 91d515f1c8d7
models/plda.binWeights199.2 KB 5b50da1827c4
models/plda.npzWeights133.9 KB 9b77bcd84069
models/powerset.binWeights84 B 1feb393d1e1f
models/powerset_mapping.npzWeights352 B 0a52d770a47f
models/xvec_transform.npzWeights134.4 KB 325f1ce8e48f
models_npu2/embtail.binWeights5.2 MB c2a1586fa656
models_npu2/fbank.binWeights83.9 KB de2fbe0c2b7d
models_npu2/fbank_params.npzWeights85.4 KB 0fa82ca61a79
models_npu2/ffn.binWeights201.2 KB d6f624b132dd
models_npu2/ffn_params.npzWeights202.7 KB 1ee1a8180e38
models_npu2/frontend.binWeights1.6 KB da0529aeb15c
models_npu2/host_emb_tail.npzWeights5.2 MB e8c1e8da0423
models_npu2/host_frontend.npzWeights4.6 KB d282e67227f6
models_npu2/lstm_params.npzWeights5.5 MB 91d515f1c8d7
models_npu2/plda.binWeights199.2 KB 5b50da1827c4
models_npu2/plda.npzWeights133.9 KB 9b77bcd84069
models_npu2/powerset.binWeights84 B 1feb393d1e1f
models_npu2/powerset_mapping.npzWeights352 B 0a52d770a47f
models_npu2/xvec_transform.npzWeights134.4 KB 325f1ce8e48f
config.jsonConfiguration3 B —
configuration.jsonConfiguration256 B —
models/model_meta.jsonConfiguration2.1 KB —
models_npu2/model_meta.jsonConfiguration2.1 KB —
python/community1_sdk/__init__.pyConfiguration186 B —
python/community1_sdk/axwrap.pyConfiguration1.7 KB —
python/community1_sdk/fbank.pyConfiguration3.2 KB —
python/community1_sdk/ffn.pyConfiguration1.4 KB —
python/community1_sdk/lstm.pyConfiguration4.7 KB —
python/community1_sdk/pipeline.pyConfiguration13.8 KB —
python/community1_sdk/vbx.pyConfiguration3.6 KB —
python/example.pyConfiguration3.9 KB —
README.mdDocumentation3.8 KB —
bin/community1_diar_ax650Other136.2 KB e0560852c461
models/embedding_convs.axmodelOther5.6 MB 711d572e2752
models/lstm_cell_l0.axmodelOther30.6 MB a4260c6faf90
models/lstm_cell_l1.axmodelOther31.1 MB 4b537cb75cfa
models/lstm_cell_l2.axmodelOther31.0 MB 6f0fa7ac0d1c
models/lstm_cell_l3.axmodelOther30.9 MB 6a00fd895889
models/segmentation_convs1.axmodelOther43.0 KB —
models/segmentation_convs2.axmodelOther34.8 KB —
models/segmentation_sincnet.axmodelOther50.0 KB —
models_npu2/embedding_convs.axmodelOther5.6 MB bc64bdbb5faf
models_npu2/lstm_cell_l0.axmodelOther30.6 MB 44a6a2c2ead7
models_npu2/lstm_cell_l1.axmodelOther31.1 MB ce8553c561fd
models_npu2/lstm_cell_l2.axmodelOther31.0 MB 8f959806337f
models_npu2/lstm_cell_l3.axmodelOther30.9 MB dbe3594fbfbd
models_npu2/segmentation_convs1.axmodelOther39.0 KB —
models_npu2/segmentation_convs2.axmodelOther31.3 KB —
models_npu2/segmentation_sincnet.axmodelOther438.0 KB db171fb319c7
requirements.txtOther44 B —
run_ax650.shOther1.0 KB —
run_cpp_ax650.shOther514 B —
samples/sample_meeting.wavOther3.8 MB 81bd9e5314f1
.gitattributesRepository2.8 KB —

License and Download

License
cc-by-4.0
Access
Open weights, no gate
Download size
34.1 MB
Download from ZY

Released by ZY through its official repository on Hugging Face. Read the license.

Memory Requirements

PrecisionWeights in memory
As published34.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

Model · Audio classification

ced-gguf

Mudler

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 weights apache-2.0 ced.cpp

Model · Audio classification

lang-id-voxlingua107-ecapa

SpeechBrain

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…

Open weights apache-2.0 speechbrain

Model · Audio classification

accent-id-commonaccent_ecapa

Juan Pablo Zuluaga

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…

Open weights mit speechbrain

Model · Audio classification

MERT-v1-330M

Multimodal Art Projection

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…

Open weights cc-by-nc-4.0 transformers

Model · Audio classification

MERT-v1-95M

Multimodal Art Projection

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…

Open weights cc-by-nc-4.0 transformers

Model · Audio classification

gender_cls_svm_ecapa_voxceleb

Gregory Koushnir

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…

Open weights apache-2.0