python使用whisper+moviepy实现视频自动识别转成srt字幕文件代码

代码语言:python

所属分类:其他

代码描述:python使用whisper+moviepy实现视频自动识别转成srt字幕文件代码,使用moviepy从视频中抽取分割音频,然后用whisper转换成文字字幕srt文本文件。

代码标签: python whisper moviepy 视频 自动 识别 转成 srt 字幕 文件 代码

下面为部分代码预览,完整代码请点击下载或在bfwstudio webide中打开

#pip install -U openai-whisper
#pip install -U opencc
#pip install moviepy
import whisper
from moviepy.editor import VideoFileClip

# 提取音频文件
def extract_audio_from_video(video_file, audio_file):
    video = VideoFileClip(video_file)
    audio = video.audio
    audio.write_audiofile(audio_file, codec='pcm_s16le')  # 使用无损的音频编码

# 使用 Whisper 转录音频并生成 SRT 文件
def generate_srt_from_audio(audio_file, srt_file):
    # 加载 Whisper 模型
    model = whisper.load_model("base")

    # 转录音频并获取时间戳
    result = model.transcribe(audio_file, language="zh", task="transcribe")

    # 导出 SRT 字幕文件
    def export_srt(result, filename):
        def srt_timestamp(seconds):
            h = int(seconds // 3600)
            m = int((seconds % 3600) // 60)
            s = int(seconds % 60)
            ms = int((seconds % 1) * 1000)
            return f"{h:02}:{m:02}:{s:02}.........完整代码请登录后点击上方下载按钮下载查看

网友评论0