连续图片转化成视频格式的脚本

代码

使用说明：执行： python this_scripts.py +参数
参数是 存图片的文件夹名字
该代码中有：统计一个文件夹下的图片数量，以及图片的size

#coding=gbk
#instructions:  figures transform to video
#              RUN: python xx.py argv 
#              the argv is the upper level of the image directory
import cv2
import os
import random
import sys

file_name = sys.argv[1]
#图片路径
im_dir = './20bn-something-something-v1/'+ file_name
#输出视频路径
video_dir = file_name +'.mp4'
#帧率
fps = 30  
#图片数
num = len(os.listdir(im_dir))
print ("the number of figures:", num)
#图片尺寸
im_first_name = os.path.join(im_dir, str(1).zfill(5)+'.jpg')
size_frame = cv2.imread(im_first_name)
img_size_first = size_frame.shape
#print("sssssssssizeeee", img_size_first)
#print("sssssssssize", img_size_first[1], img_size_first[0])
img_size = (img_size_first[1],img_size_first[0])

#fourcc = cv2.cv.CV_FOURCC('M','J','P','G')#opencv2.4
fourcc = cv2.VideoWriter_fourcc(*'mp4v') #opencv3.0

videoWriter = cv2.VideoWriter(video_dir, fourcc, fps, img_size)
for i in range(1,num):
    im_name = os.path.join(im_dir, str(i).zfill(5)+'.jpg')
    frame = cv2.imread(im_name)
    size = frame.shape
    print("figure size", size)
    videoWriter.write(frame)
    print (im_name)
videoWriter.release()
print ('finish')