最近面试,有录音,想把录音快速转文字查漏补缺,但是找了几个小程序要花钱,不如直接写个脚本之际转

(下载FFmpeg 很重要)

访问 FFmpeg 官网:https://ffmpeg.org/download.html

找到 Windows 板块,点击 "Windows builds from gyan.dev"

下载 ffmpeg-release-full.7z

并需要配置环境变量

安装python依赖

pip install openai-whisper
pip install ffmpeg-python

脚本

import whisper

model = whisper.load_model("base")
result = model.transcribe("面试2.m4a", language="zh")
print(result["text"])
with open("transcript3.txt", "w", encoding="utf-8") as f:
    f.write(result["text"])

模型精度选择(免费)

-  base :最快,普通说话清晰够用

-  small :中文识别明显更好

-  medium :接近付费平台效果

-  large :最强,但需要显卡

另外由于我的电脑是有独显的,但是whisper没有调用到,直接用的cpu,导致转录很慢

修改方式,更新pytorch为可以兼容cuda显卡的

pip uninstall -y torch torchvision torchaudio

pip install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple --extra-index-url https://download.pytorch.org/whl/cu118

更新代码

import whisper

# 加载模型,并手动移动到 GPU (cuda)
model = whisper.load_model("base").to("cuda")

# 转写时,关闭 FP16 可以避免某些兼容性警告
result = model.transcribe("面试2.m4a", language="zh", fp16=False)

print(result["text"])
with open("transcript3.txt", "w", encoding="utf-8") as f:
    f.write(result["text"])

Logo

Agent 垂直技术社区,欢迎活跃、内容共建。

更多推荐