python+opencv图片转成视频

# coding:utf-8

import os
import cv2
import re

from PIL import Image


def get_images(path):
    file_list = []
    for root, dirs, files in os.walk(path):
        if not files:
            continue
        for file in files:
            if file.endswith('.jpg'):
                file_list.append(os.path.join(root, file))
                # file_list.append(file)
    return file_list


def key_sort(image_path):
    pattern = re.compile("\d+")
    image_name = os.path.basename(image_path)
    return int(pattern.findall(image_name)[0])


def main():
    # 图片路径
    path = "images/"
    # 将图片存入列表
    file_list = get_images(path)
    img=Image.open(file_list[0])
    # 按照图片名称排序
    file_list.sort(key=key_sort)
    # 一秒4帧，代表1秒视频由4张图片组成
    fps = 6
    # 视频分辨率，图片的分辨率，切记文件夹的图片的分辨率都是一样的
    img_size = (img.width, img.height)
    # 保存视频的路径
    save_path = "test.avi"
    fourcc = cv2.VideoWriter_fourcc('M', 'J', 'P', 'G')
    video_writer = cv2.VideoWriter(save_path, fourcc, fps, img_size)
    for file_name in file_list:
        file_name
        img = cv2.imread(file_name)
        video_writer.write(img)

    video_writer.release()


if __name__ == "__main__":
    main()