【3k star!】最新开源ASR模型Moonshine性能凌驾whisper: 速率更快、正确率 ...

张裕  金牌会员 | 2025-3-18 15:34:11 | 显示全部楼层 | 阅读模式
打印 上一主题 下一主题

主题 671|帖子 671|积分 2013

媒介

本日给各人分享最新开源的一款ASR模型Moonshine, 据官方先容,比whisper划一级规模的参数版本识别速率更快,正确率更高! 本文将准备多个案例进行实测ASR效果,仅供参考。下面进入本日的主题~
本文目次



  • moonshine模型先容
  • moonshine性能先容
  • 实战篇:下载moonshine模型权重&&步骤代码进行语音识别
  • 使用 onnxruntime 包来运行 Moonshine 模型,不依靠torch
  • 使用huggingface框架加载moonshine模型进行asr语音识别
  • 效果篇: moonshine-base模型 VS whisper-base模型ASR效果对比
  • 案例1: 短文本语音识别-2种模型效果展示
  • 案例2: TED演讲视频-2种模型效果展示
moonshine模型先容

Moonshine 是由 Useful Sensors公司推出开源的语音到文本(speech-to-text, STT)转换模型,旨在为资源受限设备提供快速而正确的主动语音识别(ASR)服务。Moonshine 基于先进的编码器-解码器架构,采用了Transformer模型。其编码器部分负责处置惩罚输入的语音信号,而解码器部分则生成文本输出。如今在gitihub社区点赞量达2k!
Moonshine模型具有以下特点:

  • 开源tiny版本,参数量:27 M, 只支持英文语言; 开源base版本,参数量:61 M, 只支持英文语言;
  • 更快的处置惩罚速率,Moonshine 的处置惩罚速率比 Whisper 快 1.7 倍。对于 10 秒的短音频片段,处置惩罚速率可达 Whisper 的五倍。
  • 基于20w小时的语音样本练习而来。
moonshine性能先容

Moonshine 在多个维度上逾越了现有的语音识别办理方案,特别是在处置惩罚速率和正确度方面。据官方报告,Moonshine 的处置惩罚速率「比 OpenAI 的 Whisper 快五倍」,而且在词错误率方面也体现得更好,如下图所示。
这种显著的优势使得 Moonshine 成为资源受限环境下语音识别的理想选择。
下面我将给各人实操部署moonshine-base模型和whisper-base模型,准备几个案例来实际展示具体的语音识别效果,仅供参考~
实战篇:下载moonshine模型权重&&步骤代码进行语音识别

使用 onnxruntime 包来运行 Moonshine 模型,不依靠torch

  1. from IPython.display import clear_output   !pip install moonshine   !git clone https://github.com/usefulsensors/moonshine.git   !pip install silero_vad onnxruntime  sounddevice tokenizers einops   !pip install onnxruntime-gpu   
复制代码
下载模型权重

  1. !huggingface-cli download UsefulSensors/moonshine  --local-dir . --local-dir-use-symlinks False   clear_output()   
复制代码
加载moonshine模型onnx格式权重进行推理

  1. model = MoonshineOnnxModel(models_dir= "./onnx/base")   def moonshine_infer(wav_file):       with wave.open(wav_file) as f:           params = f.getparams()           assert (               params.nchannels == 1               and params.framerate == 16_000               and params.sampwidth == 2           ), f"wave file should have 1 channel, 16KHz, and int16"           audio = f.readframes(params.nframes)       audio = np.frombuffer(audio, np.int16) / 32768.0       audio = audio.astype(np.float32)[None, ...]       tokens = model.generate(audio)       tokenizer = tokenizers.Tokenizer.from_file("./moonshine/assets/tokenizer.json")       text = tokenizer.decode_batch(tokens)       return text   
复制代码
进行模型推理


使用huggingface框架加载moonshine模型进行asr语音识别

  1. %%time    %cd /kaggle/working/moonshine   from IPython.display import clear_output   from transformers import AutoModelForSpeechSeq2Seq, AutoConfig, PreTrainedTokenizerFast   import torchaudio   import torch   import sys   device = "cuda:0" if torch.cuda.is_available() else "cpu"   # 'usefulsensors/moonshine-base' for the base model   moonshine = AutoModelForSpeechSeq2Seq.from_pretrained('usefulsensors/moonshine-base', trust_remote_code=True)   tokenizer = PreTrainedTokenizerFast.from_pretrained('usefulsensors/moonshine-base')   audio, sr = torchaudio.load("moonshine/assets/beckett.wav")   if sr != 16000:       audio = torchaudio.functional.resample(audio, sr, 16000)   tokens = moonshine(audio)   print(tokenizer.decode(tokens[0], skip_special_tokens=True))   
复制代码

下面我将利用moonshine-base版本和whisper-base版本的模型进行语音识别效果对比,看具体实际案例情况下,模型的具体体现情况,任意找的素材,经供参考~
效果篇: moonshine-base模型 VS whisper-base模型ASR效果对比

案例1: 短文本语音识别-2种模型效果展示

参考音频1效果展示:
moonshine-base的ASR效果展示

识别结果: Ever tried ever failed, no matter try again fail again fail better.
whisper-base的ASR效果展示


2种模型识别结果都非常正确,而moonshine-base速率很快,只用来不到0.6秒,即便算上模型加载的时间,也才1.2秒。
案例2: TED演讲视频-2种模型效果展示

任意找一份英语的演讲视频进行测试,我这个找到https://www.youtube.com/playlist?list=PLosaC3gb0kGDUYoRq-VioWOZ5Ke0UIoSE,截取前2分钟的视频转化为音频效果如下:
moonshine-base的ASR效果展示

单次不支持长语音识别,采用分段识别,代码如下:
  1. import librosa   import os   import moonshine   import soundfile as sf   # !mkdir temp   def benchmark(audio_pth):       # 读取音频文件       audio, sr = torchaudio.load("english.wav")       if sr != 16000:           audio = torchaudio.functional.resample(audio, sr, 16000)       # 分割音频文件成小段       chunk_duration = 10  # 每个片段的长度(秒)       chunk_size = int(chunk_duration * sr)       chunks = [audio[0:1,i:i + chunk_size] for i in range(0, audio.shape[1], chunk_size)]          # 转录音频       transcription = ""       for i, chunk in enumerate(chunks):           print(f"正在转录... ({i + 1}/{len(chunks)})")           tokens = moonshine_model(chunk)           chunk_transcription = tokenizer.decode(tokens[0], skip_special_tokens=True)              if isinstance(chunk_transcription, list):               chunk_transcription = ' '.join(chunk_transcription)              transcription += chunk_transcription       return transcription   
复制代码
最后识别的结果如下:

whisper-base的ASR效果展示


我感觉2个模型识别的效果相差不大,moonshine速率是比较快的,但是如今moonshine只支持英文。各人可以对比录音听听,看看谁识别的更准~
如何学习AI大模型 ?

“最先掌握AI的人,将会比较晚掌握AI的人有竞争优势”。

这句话,放在计算机、互联网、移动互联网的开局时期,都是一样的道理。
我在一线互联网企业工作十余年里,引导过不少同行后辈。资助许多人得到了学习和发展。
我意识到有许多履历和知识值得分享给各人,故此将并将重要的AI大模型资料包罗AI大模型入门学习思维导图、佳构AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。【保证100%免费】

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

张裕

金牌会员
这个人很懒什么都没写!
快速回复 返回顶部 返回列表