当我们使用opencv-python利用图像生成视频时,通常需要cv2.VideoWriter()命令,比如下面给出一个具体例子

“”“
下面程序是将img_in_dir目录下的100帧图像转成视频video.mp4
”“”
import os
import cv2
import numpy as np

def visualize(img_in_dir, video_out_path):
    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
    out = cv2.VideoWriter(filename=video_out_path, fourcc=fourcc, fps=30, frameSize=(1920, 1080))
    for frame_id in np.arange(1, 100):
        print('processing frame {}...'.format(frame_id))
        img_path = os.path.join(img_in_dir, 'frame_%05d.jpg'%frame_id)
        img = cv2.imread(img_path)
        out.write(img)
    out.release()

if __name__ == '__main__':
    img_in_dir = r'D:\Data\ActEV\train_frame\VIRAT_S_000000'
    video_out_path = r'D:\Data\ActEV\train_frame\video.mp4'
    img = cv2.imread(r'D:\Data\ActEV\train_frame\VIRAT_S_000000\frame_00001.jpg')
    visualize(img_in_dir, video_out_path)

其中,cv2.VideoWriter()中指定了输出视频的一些参数,如编码、帧率、分辨率等。
这里需要注意,分辨率需要保证和图像分辨率一致,否则会导致输出视频无效(通常只有1KB)

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐