这是一款python源码,实现将没有文字字幕的视频,通过语音识别技术识别为文本,然后重写将字幕添加到视频底部的效果代码。
注意调用的是google语音识别服务,确保你的网络能访问谷歌。
import os import speech_recognition as sr from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip from pydub import AudioSegment def extract_audio(video_path): video = VideoFileClip(video_path) audio = video.audio audio.write_audiofile("temp_audio.wav") return "temp_audio.wav" def transcribe_audio(audio_path): recognizer = sr.Recognizer() audio = AudioSegment.from_wav(audio_path) # 将音频分割成较小的片段 chunk_length_ms = 10000 # 10秒 chunks = [audio[i:i+chunk_length_ms] for i in range(0, len(audio), chunk_length_ms)] transcriptions = [] for i, chunk in enumerate(chunks): chunk.export(f"temp_chunk_{i}.wav", format="wav") with sr.AudioFile(f"temp_chunk_{i}.wav") as source: audio_data = recognizer.record(source) try: text = recognizer.recognize_google(audio_data, language="zh-CN") transcriptions.append((i * chunk_length_ms / 1000, text)) except sr.UnknownValueError: print(f"Could not understand audio in chunk {i}") except sr.RequestError as e: print(f"Could not request results from Google Speech Recognition service; {e}") os.remove(f"temp_chunk_{i}.wav") return transcriptions def create_subtitle_clips(transcriptions, videosize, fontsize=24, font='Arial', color='white', method='caption'): subtitle_clips = [] for i, (t, text) in enumerate(transcriptions): if i < len(transcriptions) - 1: duration = transcriptions[i+1][0] - t else: duration = 5 # 最后一个字幕显示5秒 text_clip = TextClip(text, fontsize=fontsize, font=font, color=color, size=videosize, method=method) text_clip = text_clip.set_pos('bottom').set_duration(duration).set_start(t) subtitle_clips.append(text_clip) return subtitle_clips def add_subtitles_to_video(video_path, output_path): # 提取音频 audio_path = extract_audio(video_path) # 转写音频 transcriptions = transcribe_audio(audio_path) # 创建视频剪辑 video = VideoFileClip(video_path) # 创建字幕剪辑 subtitle_clips = create_subtitle_clips(transcriptions, video.size) # 将字幕添加到视频 final_video = CompositeVideoClip([video] + subtitle_clips) # 写入新视频 final_video.write_videofile(output_path) # 清理临时文件 os.remove(audio_path) # 使用示例 input_video = "inputvideo.mp4" output_video = "output.mp4" add_subtitles_to_video(input_video, output_video)
网友回复
js如何流式输出ai的回答并折叠代码块,点击代码块右侧可预览代码?
ai大模型如何将文章转换成可视化一目了然的图片流程图图表?
大模型生成html版本的ui原型图和ppt演示文档的系统提示词怎么写?
rtsp视频直播流如何转换成websocket流在h5页面上观看?
为啥coze会开源工作流agent coze studio?
如何检测网页是通过收藏夹打开的?
python如何实现类似php的http动态脚本请求处理响应代码?
js如何实现类似php的http动态脚本请求处理响应代码?
trae与solo有啥区别不同?
vue如何让ai动态生成问卷调查多步骤表单式收集基础信息自动规划执行任务?