最近考虑将深度模型在视频帧上的预测结果直接写入视频中，采用cv2中的VideoWriter写视频，发现生成的视频都只有几KB。
查询了各种方案，最大的可能是写入视频的size和图像的size不匹配。但是按照我的代码是直接获取的原视频的size，一般不会有这个问题。
多方查看分析，发现是模型输出的图像size和原视频的size差了几个像素，导致了这种不匹配。
实际使用时，如果出现了类似的问题，可以把模型输出的图像保存下来，对比原视频的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可能会有不匹配
while cap.isOpened():
ret, frame = cap.read()
if ret:
result = model(frame) # 深度模型输出的图像
video.write(result) # 写入视频
else:
break
cap.release()
video.release()
