python通过cv2中VideoWriter输出的视频文件只有几KB的问题解决

最近考虑将深度模型在视频帧上的预测结果直接写入视频中，采用cv2中的VideoWriter写视频，发现生成的视频都只有几KB。

查询了各种方案，最大的可能是写入视频的size和图像的size不匹配。但是按照我的代码是直接获取的原视频的size，一般不会有这个问题。

多方查看分析，发现是模型输出的图像size和原视频的size差了几个像素，导致了这种不匹配。

实际使用时，如果出现了类似的问题，可以把模型输出的图像保存下来，或者print(image.shape)来查看对比原视频的size，看二者是否匹配。如果不匹配的话，需要将cv2.VideoWriter中的size设置为实际输出的图像size。

实现的代码如下（省略了深度模型输出图像的后处理部分）：

# 输入视频
filePath = 'test.mp4'
cap = cv2.VideoCapture(filePath)
fps = cap.get(cv2.CAP_PROP_FPS)
width = cap.get(cv2.CAP_PROP_FRAME_WIDTH)
height = cap.get(cv2.CAP_PROP_FRAME_HEIGHT)

size = (int(width), int(height))

# 设置编码格式、输出路径、fps、size
fourcc = cv2.VideoWriter_fourcc('m', 'p', '4', 'v')
video = cv2.VideoWriter('output.avi', fourcc, fps, size)  # size可能会与result尺寸不匹配

while cap.isOpened():
	ret, frame = cap.read()
	if ret:
		result = model(frame)  # 深度模型输出的图像
        # print(result.shape)
		video.write(result)  # 写入视频
	else:
		break

cap.release()
video.release()