opencv-python: 输出视频——正确设置分辨率

当我们使用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）