近来有一个音频转文本的需求,相识到了OpenAI的whisper模型可以实现。
Whisper 是 OpenAI 提供的一个通用语音识别模型,支持多语言的音频转文本功能,并具有较高的准确性。它的主要用途包括自动语音识别 (ASR)、语言翻译(将音频直接翻译成英文文本)等。Whisper 支持将长时间音频文件(如对话、采访、演讲)转录成笔墨,而且在多语言环境下体现良好。
Whisper 的核心功能
- 音频转文本:自动将音频内容转录为文本。
- 多语言支持:Whisper 支持包括中文、法语、西班牙语、阿拉伯语等多种语言。
- 自动语言检测:能够识别音频语言,无需提前指定。
- 翻译:将非英文音频直接翻译成英文文本。
Whisper 模型的安装和使用
1. 下载 Whisper
- git clone https://github.com/openai/whisper.git
- cd whisper
- python -m venv myvenv #创建虚拟环境
- venv\Scripts\activate #激活虚拟环境
- pip install -r requirements.txt #在该虚拟环境下安装依赖库
复制代码 2. 安装 ffmpeg
自行参考Windows安装安装ffmpeg
如果ffmpeg没有安装成功,运行代码会报如下错误:
我一开始碰到这个题目以为是音频文件的路径错了
3. 使用 Whisper 举行音频转文本
Whisper 提供了简单的 API,可以很方便地加载模型并举行转录。以下是一个基本的代码示例:
- import whisper
- from pathlib import Path
- import json
- class TextSegment:
- def __init__(self, text, begin, end):
- self.text = text
- self.begin = begin
- self.end = end
- def to_dict(self):
- # 将对象转换为字典
- return {
- "text": self.text,
- "begin": self.begin,
- "end": self.end
- }
- def __repr__(self):
- return f'TextSegment(text="{self.text}", begin={self.begin}, end={self.end})'
- # 加载模型(可以选择不同大小的模型,越大准确率越高,但速度较慢)
- model = whisper.load_model("tiny") # 可选 "tiny", "base", "small", "medium", "large"
- # 转写音频文件
- path = Path("path/A.mp3")
- result = model.transcribe(str(path)) # 替换为你的音频文件路径
- print(result)
- TextSegmentList = []
- for item in result['segments']:
- TextSegmentList.append(TextSegment(item["text"], item["start"], item["end"]))
- print(f"{item['text']}-----{item['start']}----{item['end']}")
- json_string = json.dumps([segment.to_dict() for segment in TextSegmentList], indent=4)
- print(json_string)
- with open(path.stem+'.txt', 'w') as file:
- file.write(json_string)
复制代码 模型返回的数据布局如下所示:
数据布局有文本,以及对应的结束和开始的时间,有了这些数据无论是生成视频字幕还是音频和文本同步定位都是很方便的
4. 模型选择
Whisper 提供多种大小的模型,分别为 tiny、base、small、medium 和 large。模型越大,识别的准确性越高,但计算资源需求也会增加。可以根据需要选择符合的模型。
5. Whisper 的源码布局
Whisper 的代码布局清晰,主要包罗以下核心部门:
- whisper/model.py:定义模型布局和加载逻辑。
- whisper/transcribe.py:处置惩罚音频文件转录的逻辑。
- whisper/tokenizer.py:将音频编码成模型可以明确的 token,并将转录后的 token 转换为文本。
- whisper/audio.py:负责音频文件的读取和预处置惩罚。
6. 应用场景
Whisper 实用于多种音频转录场景,如会议记录、字幕生成、采访转录、实时语音识别等。其多语言支持让它可以应对多语言环境的音频内容。
7. 性能优化建议
Whisper 模型对计算资源要求较高,建议在 GPU 环境下运行,并根据需求选择较小的模型(如 tiny 或 base),试过large模型电脑直接卡死。音频质量越高,转录的效果越好,噪声较大的音频会影响识别效果。
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。 |