【脚本】图片-音视频-压缩文件处理

2023-12-14 09:40:26

提前导入模块

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.shape

    fourcc = 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_length
        end = start + segment_length
        chunk = 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 Image
    image = 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)}")

文章来源:https://blog.csdn.net/qq_45437316/article/details/134859178
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。