Whisper:从 3900 万参数到 15.5 亿参数,六种尺寸按显存和精度挑

Whisper 是那种「一个模型顶一条流水线」的项目。传统语音处理要分好几步走:先做语种判断,再做识别,需要翻译还得再接一个翻译模型,语音活动检测往往又是单独的一环。Whisper 把这些任务统一成解码器要预测的标记序列,用一个 Transformer 编码解码结构一起学出来。

官方在仓库里给出的多任务训练数据规模标注是 68 万小时,训练素材覆盖多种语言与不同录音条件。项目的定位写得很直接——通用的语音识别模型。

Whisper 官方仓库的项目说明与 Approach 架构图

一个模型做完几件事

官方把能力列成四项:多语言语音识别、语音翻译、口语语种识别和语音活动检测。翻译指的是把非英语语音直接转成英语文本,而不是任意语种之间互译,这一点在实际使用时要先明确。

处理长音频的方式是滑动窗口。官方在文档里说明,转写方法会把整个文件读进来,然后以 30 秒为一个窗口逐段做自回归预测。也就是说,一个小时的录音会被切成一百多个窗口连续处理,不需要你自己先去切片。

六种尺寸怎么挑

Whisper 官方仓库的模型尺寸、参数、显存与相对速度对照表

这是选型时最该看的一张表。官方提供六个尺寸,参数从 tiny 的约 3900 万一直到 large 的约 15.5 亿,显存需求从约 1GB 到约 10GB。相对速度那一列是以 large 为基准测出来的,tiny 约为它的 10 倍,turbo 约为 8 倍,可以看到速度与体积之间基本是一条平滑的取舍曲线。

如果显存紧张,tiny 和 base 的显存需求都只有约 1GB,跑在普通笔记本上也不算勉强;要更好的准确率就得往 medium 和 large 走。中间那个 turbo 值得单独提一下:官方说明它是 large-v3 的优化版本,用极小的准确率损失换来了明显更快的转写速度,算是一个折中选项。

另外,tiny、base、small、medium 这四个尺寸还额外提供了仅英文版本的模型。官方观察到,纯英文任务下这些版本往往表现更好,不过在参数越大的尺寸上这个差距越不明显。

装起来:pip 加一个 ffmpeg

Whisper 官方仓库的安装说明与 ffmpeg 依赖

安装本身只要一条 pip 命令,真正需要留意的是系统依赖:Whisper 依赖 ffmpeg 这个命令行工具来做音频解码。官方按不同系统列出了安装方式,常见的包管理器都覆盖了,照着装即可。

还有一处需要留意:官方提醒,如果你的平台没有现成的分词器预编译包,pip 安装过程中会需要 Rust 环境;报错信息里出现找不到某个模块时,补装一个对应的包就能解决。环境版本上,代码库兼容 Python 3.8 到 3.11 以及较新的 PyTorch 版本。

两种用法:命令行与 Python

Whisper 官方仓库的命令行用法与 turbo 模型说明

命令行是最快的验证方式:把音频文件跟在命令后面,用 --model 指定模型就能转写;如果音频不是英语,用 --language 指定语种;要做翻译则加上 --task 指定翻译任务。

这里有一条限制需要特别记住——官方明确说明 turbo 模型没有为翻译任务做过训练。即使你指定了翻译任务,它仍然会返回原语言的内容。要用翻译功能,得换回 tiny、base、small、medium 或 large 这些多语言模型,官方建议翻译场景用 medium 或 large 效果更好。

Whisper 官方仓库的 Python 调用示例

Python 里的调用同样简短:加载模型、把音频路径交给转写方法、取出结果里的文本字段,三步就够。官方还提供了更底层的接口,可以自己加载音频、生成对数梅尔频谱图、单独调用语种检测或解码,方便把它嵌进更复杂的流程里。

许可是这个项目的一个加分项

Whisper 的代码和模型权重都以 MIT 许可证发布,商用限制很少。这一点在语音识别这一类项目里并不常见——不少模型的权重会附带单独的、更严格的许可条款。如果你需要把语音识别能力放进自己的产品,这一条能省掉不少法务上的沟通成本。

它适合谁

如果你要给视频批量做字幕、把会议录音整理成文字、或者给自己的应用加一个本地转写功能,Whisper 的上手成本很低:一条命令装好、一条命令跑起来,六种尺寸可以先从小的试,效果不够再换大的。加上 MIT 许可,落地时不需要为授权问题反复确认。

需要注意的是它并非全能:翻译只支持转成英语这一个方向,实时性要求高的场景要评估自己的硬件能不能跟上,嘈杂环境下的准确率也和录音质量直接相关。把它当成一个可靠的离线转写工具,预期会比较稳。

开源加宽松许可,是 Whisper 被大量项目当作默认选择的原因。工具本身的门槛已经很低,剩下的差别往往在于你喂给它的音频有多干净。

Whisper下载地址
支持的操作系统: Windows Linux macOS