音视频处理
- 一,图片操作
- 1,转换图片格式
- 2,多张图片合成视频
- 二,音频操作
- 1,转换音频格式
- 2,分割音频为多段
- 3,合成多段音频
- 三,视频操作
- 1,转换视频格式
- 2,提取视频中的音频
- 3,合成多段视频
- 4,为视频添加音频
- 5,视频抽帧
- 四,压缩文件
- 1,Zip转rar
- 2,rar转Zip
提前导入模块
from pydub import AudioSegment
from moviepy.editor import VideoFileClip, concatenate_videoclips, AudioFileClip
from PIL import Image
import cv2
import os
一,图片操作
1,转换图片格式
def convert_image(input_path, output_path, output_format):"""将图片从一种格式转换为另一种格式。:param input_path: 输入图片文件的路径。:param output_path: 输出图片文件的路径。:param output_format: 输出图片的格式(如 'PNG', 'JPEG' 等)。"""try:# 打开输入图片image = Image.open(input_path)# 将图片保存为指定的输出格式image.save(output_path, format=output_format)print(f"成功将图片转换为 {output_format} 格式并保存到 {output_path}")except Exception as e:print(f"转换图片格式时出现错误: {str(e)}")
2,多张图片合成视频
def images_to_video(image_folder, output_video_path, frame_rate=30):"""将多张图片合成为视频。:param image_folder: 包含要合成的图片的文件夹路径。:param output_video_path: 输出视频的文件路径。:param frame_rate: 视频的帧率(默认为30帧/秒)。"""image_files = [img for img in os.listdir(image_folder) if img.endswith(".png") or img.endswith(".jpg")]image_files.sort() # 确保图片按顺序排列frame = cv2.imread(os.path.join(image_folder, image_files[0]))height, width, layers = frame.shapefourcc = cv2.VideoWriter_fourcc(*'mp4v') # 选择视频编解码器(可以根据需要更改)video = cv2.VideoWriter(output_video_path, fourcc, frame_rate, (width, height))for image_file in image_files:image_path = os.path.join(image_folder, image_file)frame = cv2.imread(image_path)video.write(frame)video.release()print(f"成功生成视频:{output_video_path}")
二,音频操作
1,转换音频格式
def convert_audio_format(input_path, output_path, output_format):"""将音频文件从一种格式转换为另一种格式。:param input_path: 输入音频文件的路径。:param output_path: 输出音频文件的路径。:param output_format: 输出音频的格式(如 'mp3', 'wav' 等)。"""# 加载音频文件audio = AudioSegment.from_file(input_path)# 转换并导出音频audio.export(output_path, format=output_format)print(f"音频已从 {input_path} 转换为 {output_format} 格式,并保存到 {output_path}。")
2,分割音频为多段
def split_audio_into_chunks(audio_path, format,segment_length=10000):"""将音频文件分割成多个片段。:param audio_path: 音频文件的路径。:param segment_length: 分割长度,单位为毫秒。默认为 10000 毫秒(10秒)。:return: 分割后的音频片段列表。"""# 加载音频文件audio = AudioSegment.from_file(audio_path)# 计算分割数量length_of_audio = len(audio)number_of_chunks = length_of_audio // segment_length + (1 if length_of_audio % segment_length else 0)# 分割音频chunks = []for i in range(number_of_chunks):start = i * segment_lengthend = start + segment_lengthchunk = audio[start:end]chunks.append(chunk)chunk.export(f"chunk{i}.mp3", format) # 导出每个片段为单独的文件return chunks
3,合成多段音频
def merge_audio_files(audio_files, output_path):"""合并多个音频文件为一个音频文件。:param audio_files: 音频文件路径列表。:param output_path: 合并后的音频文件输出路径。"""# 初始化空的音频段combined = AudioSegment.empty()# 依次加载每个音频文件并合并for file in audio_files:audio = AudioSegment.from_file(file)combined += audio# 导出合并后的音频文件combined.export(output_path, format="mp3")print(f"音频文件已合并并保存到 {output_path}")
三,视频操作
1,转换视频格式
def convert_video_format(input_path, output_path, output_format):"""将视频文件从一种格式转换为另一种格式。:param input_path: 输入视频文件的路径。:param output_path: 输出视频文件的路径。:param output_format: 输出视频的格式(如 'mp4', 'avi' 等)。"""# 加载视频文件video = VideoFileClip(input_path)# 设置输出文件的扩展名output_path_with_format = output_path + '.' + output_format# 转换并导出视频video.write_videofile(output_path_with_format, codec='libx264')print(f"视频已从 {input_path} 转换为 {output_format} 格式,并保存到 {output_path_with_format}。")
2,提取视频中的音频
def extract_audio_from_video(video_path, audio_output_path):"""从视频文件中提取音频。:param video_path: 视频文件的路径。:param audio_output_path: 要保存的音频文件的路径。"""# 加载视频文件video = VideoFileClip(video_path)# 提取音频audio = video.audio# 保存音频文件audio.write_audiofile(audio_output_path)print(f"音频已从 {video_path} 提取并保存到 {audio_output_path}。")
3,合成多段视频
def merge_video_files(video_files, output_path):"""合并多个视频文件为一个视频文件。:param video_files: 视频文件路径列表。:param output_path: 合并后的视频文件输出路径。"""# 加载视频文件clips = [VideoFileClip(file) for file in video_files]# 合并视频final_clip = concatenate_videoclips(clips)# 导出合并后的视频文件final_clip.write_videofile(output_path, codec="libx264")print(f"视频文件已合并并保存到 {output_path}")
4,为视频添加音频
def add_audio_to_video(video_path, audio_path, output_path):"""将音频文件添加到视频文件中。:param video_path: 视频文件的路径。:param audio_path: 音频文件的路径。:param output_path: 合成后的视频文件输出路径。"""# 加载视频和音频video_clip = VideoFileClip(video_path)audio_clip = AudioFileClip(audio_path)# 将音频添加到视频中video_with_audio = video_clip.set_audio(audio_clip)# 导出最终视频文件video_with_audio.write_videofile(output_path, codec="libx264")print(f"音频已添加到视频中,并保存到 {output_path}")
5,视频抽帧
def extract_frame_from_video(video_path, time, output_image_path):"""从视频中抽取特定时间点的帧。:param video_path: 视频文件的路径。:param time: 抽取帧的时间点(秒)。:param output_image_path: 输出图像的路径。"""# 加载视频文件video_clip = VideoFileClip(video_path)# 获取特定时间点的帧frame = video_clip.get_frame(time)# 将帧保存为图像from PIL import Imageimage = Image.fromarray(frame)image.save(output_image_path)print(f"从 {video_path} 抽取的帧已保存到 {output_image_path}")
四,压缩文件
1,Zip转rar
def convert_zip_to_rar(zip_file_path, rar_file_path):"""将zip文件转换为rar文件。:param zip_file_path: 输入的zip文件路径。:param rar_file_path: 输出的rar文件路径。"""try:# 打开输入的zip文件with zipfile.ZipFile(zip_file_path, 'r') as zipf:# 创建rar文件with rarfile.RarFile(rar_file_path, 'w') as rarf:# 逐个将zip文件中的文件添加到rar文件中for file_name in zipf.namelist():with zipf.open(file_name) as zip_file:rarf.writestr(file_name, zip_file.read())print(f"成功将 {zip_file_path} 转换为 {rar_file_path}")except Exception as e:print(f"转换文件格式时出现错误: {str(e)}")
2,rar转Zip
def convert_rar_to_zip(rar_file_path, zip_file_path):"""将RAR文件转换为ZIP文件。:param rar_file_path: 输入的RAR文件路径。:param zip_file_path: 输出的ZIP文件路径。"""try:# 打开输入的RAR文件with rarfile.RarFile(rar_file_path, 'r') as rarf:# 创建ZIP文件with zipfile.ZipFile(zip_file_path, 'w', zipfile.ZIP_DEFLATED) as zipf:# 逐个将RAR文件中的文件添加到ZIP文件中for file_name in rarf.namelist():with rarf.open(file_name) as rar_file:zipf.writestr(file_name, rar_file.read())print(f"成功将 {rar_file_path} 转换为 {zip_file_path}")except Exception as e:print(f"转换文件格式时出现错误: {str(e)}")