python把文件夹中所有的图片按顺序转换成视频

最近需要把文件夹中多个图片转换成视频，试了一些软件发现都不好使。刚好最近又接触了 python+opencv的相关知识，索性参考网上代码，自己写了一个。发现才十几行。。果然人生苦短，我用python.

前面都是自定义的各种内容，最后两个循环处理就OK了。读文件夹中所有图片，然后写入视频中即可。。

import cv2
import os

# 读取文件夹的路径，这里处理多个文件夹
IMG_DIR_1 = '/home/xilinx/jupyter_notebooks/skynet_final/dataset_training/boat5/'   
IMG_DIR_2 = '/home/xilinx/jupyter_notebooks/skynet_final/dataset_training/riding4/'
IMG_DIR_3 = '/home/xilinx/jupyter_notebooks/skynet_final/dataset_training/car11/'
IMG_DIR_4 = '/home/xilinx/jupyter_notebooks/skynet_final/dataset_training/truck1/'
out_put_video_name = 'output.avi' #输出视频的名称，

IMG_DIR = [IMG_DIR_1, IMG_DIR_2, IMG_DIR_3, IMG_DIR_4]
file_type = '.jpg' #图片后缀名为 .jpg，可以调整
fps = 60    #视频的FPS，可以适当调整
size=(640,360) # 输出视频的分辨率
fourcc = cv2.VideoWriter_fourcc(*'MJPG')

for dir_num in range(0,len(IMG_DIR)):
    IMG_DIR_temp = IMG_DIR[dir_num]
    out_video_file_name = IMG_DIR_temp+out_put_video_name #设置视频的输出路径为 原文件夹下+前面设置的输出视频的名称
    if os.path.exists(out_video_file_name):
    	os.remove(out_video_file_name)
    videoWriter = cv2.VideoWriter(out_video_file_name,fourcc,fps,size)#最后一个是保存图片的尺寸
    image_list = [f for f in os.listdir(IMG_DIR_temp) if f.endswith(file_type)]
    image_list.sort(key= lambda x:int(x[:-4]))
    for i in range(0, len(image_list)):
        frame = cv2.imread(IMG_DIR_temp+image_list[i])
        videoWriter.write(frame)
    videoWriter.release()