【Python】图像批量合成视频，并以文件夹名称命名合成的视频

一个文件夹中有多个子文件夹，子文件夹中有多张图像。如何把批量把子文件夹中的图像合成视频，视频名称是子文件夹的名称，生成的视频保存到指定文件夹，效果记录。

代码

import os
import cv2


def create_video_from_images(image_folder, output_folder, fps=30):
    # 获取子文件夹名称，这将作为视频文件的名称
    video_name = os.path.basename(image_folder)
    video_path = os.path.join(output_folder, f"{video_name}.mp4")

    # 获取子文件夹中的所有图像文件
    images = [img for img in os.listdir(image_folder) if img.endswith((".png", ".jpg", ".jpeg", ".bmp", ".tiff"))]

    # 确保图像列表不为空
    if not images:
        print(f"No images found in {image_folder}")
        return

    # 对图像文件进行排序，确保它们按照正确的顺序合成视频
    images.sort()

    # 读取第一张图像以获取帧尺寸
    frame = cv2.imread(os.path.join(image_folder, images[0]))
    height, width, layers = frame.shape

    # 定义编码器并创建VideoWriter对象
    fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # 使用mp4编码
    video = cv2.VideoWriter(video_path, fourcc, fps, (width, height))

    # 遍历所有图像并将它们写入视频文件
    for image_file in images:
        image_path = os.path.join(image_folder, image_file)
        video.write(cv2.imread(image_path))

    # 释放VideoWriter并关闭所有窗口
    video.release()
    cv2.destroyAllWindows()
    print(f"Video {video_path} created successfully.")


# 主函数，用于遍历主文件夹中的所有子文件夹并创建视频
def main(main_folder, output_folder):
    # 确保输出文件夹存在，如果不存在则创建它
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    # 遍历主文件夹中的所有子文件夹
    for subdir, _, files in os.walk(main_folder):
        # 检查子文件夹中是否包含图像文件
        if any(file.endswith((".png", ".jpg", ".jpeg", ".bmp", ".tiff")) for file in files):
            create_video_from_images(subdir, output_folder)


# 设置主文件夹和输出文件夹的路径
main_folder_path = "F:/test_video/image/"
output_folder_path = "F:/test_video/video/"

# 调用主函数开始处理
main(main_folder_path, output_folder_path)