cv2图像转视频报错

今天在尝试用python+opencv将一些图片转换成视频, 然而怎么都转不过来, 每次播放视频都会出现播放错误.

怀疑是图片的尺寸出了问题, 然而把所有的图片size打出来看, 发现都是一样的. 最后发现, 我获取图片尺寸的时候用的是im.shape[:2], 然后直接将这个这个尺寸输入到了videowriter中, 有可能是宽和高反了, 写入的图片都是错的. 之后把videowriter中size的宽和高调换一下,果然就成功了.
代码:

import os
import cv2

img_root = 'im_path'
video_root = 'video_path'
img_list = os.listdir(img_root)
img_list.sort()
im_size = cv2.imread(os.path.join(img_root, img_list[0])).shape[:2]
fps = 24

fourcc = cv2.VideoWriter_fourcc(*'MJPG')
# 这里要调换宽和高的顺序
videoWriter = cv2.VideoWriter(os.path.join(video_root, 'video1.avi'), fourcc, fps, (im_size[1], im_size[0])) 
for i, im_name in enumerate(img_list):
    frame = cv2.imread(os.path.join(img_root, im_name))
    videoWriter.write(frame)