File size: 2,083 Bytes
139f25e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | """WakeUp.axera「你好，爱芯」唤醒词检测 Demo——复制即用。
在 AX650/AX630 板上运行：
bash setup.sh && bash run.sh
输入为 16kHz 单声道 int16 PCM（512 样本/32ms 一帧），输出逐帧唤醒得分与触发结果。
"""
import os
import sys
import argparse
from pathlib import Path
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
try:
import axengine # noqa: F401
AX_AVAILABLE = True
except Exception:
AX_AVAILABLE = False
ROOT = Path(__file__).resolve().parents[1]
MODEL = ROOT / "models" / "ax650" / "model.axmodel"
POS_WAV = ROOT / "models" / "sample_nihao_aixin.wav"
NEG_WAV = ROOT / "models" / "sample_nihao_qita.wav"
THRESHOLD = 0.615 # DET_MODE_95；DET_MODE_90 用 0.608
def main():
ap = argparse.ArgumentParser(description="WakeUp.axera 唤醒词检测 Demo")
ap.add_argument("--model", default=str(MODEL),
help="axmodel 路径（默认 models/ax650/model.axmodel；AX620E 用 models/ax620e/model.axmodel）")
args = ap.parse_args()
if not AX_AVAILABLE:
print("当前主机没有 AX 芯片（pyaxengine 不可用），无法运行 NPU 推理。")
print("本交付包为 NPU 专用版，请在 AX650/AX630/AX620E 板端执行：")
print(" bash setup.sh && bash run.sh")
return
import soundfile as sf
from wakeup_axera_sdk.detector import detect_stream
from wakeup_axera_sdk.inference import ModelSession
session = ModelSession(args.model)
for wav, expect in ((POS_WAV, "应触发"), (NEG_WAV, "应静默")):
data, sr = sf.read(str(wav), dtype="int16")
res = detect_stream(session, data, det_win=3, threshold=THRESHOLD)
status = "TRIGGERED" if res["triggered"] else "silent"
print(f"{wav.name}: max_score={res['max_score']:.3f} "
f"max_sum3={res['max_sum']:.3f} {status}（{expect}）")
print(f"阈值: {THRESHOLD}（3 帧滑动和） backend: {session.backend}")
print("✅ WakeUpModel 唤醒词检测运行成功")
if __name__ == "__main__":
main()
|