语音识别(whisper摆设)

打印 上一主题 下一主题

主题 931|帖子 931|积分 2793

whisper摆设
地址:https://github.com/openai/whisper?tab=readme-ov-file
我们使用 Python 3.9.9 和PyTorch 1.10.1 来练习和测试我们的模型,但代码库预计与 Python 3.8-3.11 和最新的 PyTorch 版本兼容。代码库还依赖于一些 Python 包,最著名的是OpenAI 的 tiktoken,用于快速标记器实现。您可以使用以下下令下载并安装(或更新到)最新版本的 Whisper:
  1. pip install -U openai-whisper
复制代码
大概,以下下令将今后存储库中提取并安装最新的提交及其 Python 依赖项:
  1. pip install git+https://github.com/openai/whisper.git
复制代码
要将包更新至此存储库的最新版本,请运行:
  1. pip install --upgrade --no-deps --force-reinstall git+https://github.com/openai/whisper.git
复制代码
它还需要ffmpeg在你的体系上安装下令行工具,大多数包管理器都可以提供该工具:
  1. # on Ubuntu or Debian
  2. sudo apt update && sudo apt install ffmpeg
  3. # on Arch Linux
  4. sudo pacman -S ffmpeg
  5. # on MacOS using Homebrew (https://brew.sh/)
  6. brew install ffmpeg
  7. # on Windows using Chocolatey (https://chocolatey.org/)
  8. choco install ffmpeg
  9. # on Windows using Scoop (https://scoop.sh/)
  10. scoop install ffmpeg
复制代码
可用型号和语言

有六种模型大小,其中四种只有英语版本,提供速度和正确性的权衡。以下是可用模型的名称及其相对于大型模型的近似内存要求和推理速度。以下相对速度是通过在 A100 上转录英语语音来测量的,实际速度可能会因多种因素而有很大差异,包罗语言、说话速度和可用的硬件。
尺寸参数纯英语模式多语言模型所需 VRAM相对速度微小的三十九 米tiny.entiny约 1 GB~10X根据74 米base.enbase约 1 GB~7倍小的244 米small.ensmall约 2 GB~4倍中等的769 米medium.enmedium约 5 GB~2倍大的1550 米不适用large~10 GB1x涡轮809 米不适用turbo约 6 GB~8倍 中文的话建议turbo模型大概large模型
以下下令将使用turbo模型转灌音频文件中的语音:
  1. whisper audio.flac audio.mp3 audio.wav --model turbo
复制代码
  1. import whisper
  2. import torch
  3. import logging
  4. # 设置日志配置,记录时间和信息
  5. logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(message)s')
  6. logging.info('============================')
  7. # 检查并设置设备
  8. device = "cuda" if torch.cuda.is_available() else "cpu"
  9. logging.info(f"Using device: {device}")
  10. # 加载模型到 GPU
  11. model = whisper.load_model("turbo", device=device)
  12. # 加载音频并将其转换为张量,并转移到 GPU
  13. audio = whisper.load_audio("output_combined.wav")
  14. audio_tensor = torch.tensor(audio).to(device)  # 将 numpy 数组转换为 PyTorch 张量并移动到 GPU
  15. # 转录音频文件
  16. result = model.transcribe(audio_tensor)
  17. # 打印转录结果并记录日志
  18. logging.info("Transcription result: %s", result["text"])
复制代码
这里建议使用GPU目前测试4090D是cpu的3呗
注意:如果电脑没有装过torch就用官方文档建议的摆设,如果有建议使用gpu
如果翻译中文使用最大的两个模型都行.比较废资源,显卡越好结果越好.

实际结果,我租的4090d显卡识别2分钟灌音转笔墨大概速度18秒左右吧,应该还可以优化.

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。

本帖子中包含更多资源

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

x
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

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

本版积分规则

梦见你的名字

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