python将图片合成为视频

#coding:utf-8

import os
import cv2
import re

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 = "./desktop/test"
	#将图片存入列表
    file_list = get_images(path)
	#按照图片名称排序
    file_list.sort(key=key_sort)
	#一秒25帧，代表1秒视频由25张图片组成
    fps = 25
	#视频分辨率
    img_size = (1920, 1080) 
	#保存视频的路径
    save_path = "./desktop/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:
        print file_name
        img = cv2.imread(file_name)
        video_writer.write(img)

    video_writer.release()

if __name__ == "__main__":
    main()