python 图片生成视频_Python：从处理过的图像中创建视频

我有一个录制汽车前视图的视频 . 该文件是.mp4，我想处理单个图像，以便我可以提取更多信息(对象，车道线等) . 问题是，当我想从处理过的文件中创建一个视频时，我收到一个错误 . 这是我到目前为止所做的：

用cv2.VideoCapture()打开视频 - 工作正常

用cv2.imwrite()保存了视频的单帧 - 工作正常

用cv2.VideoWriter()创建单帧视频 - 工作正常

使用cv2.cvtColor()，cv2.GaussianBlur()和cv2.Canny()对视频进行后处理 - 工作正常

从处理过的图像中创建视频 - 不起作用 .

这是我使用的代码：

enter code here

def process_image(image):

gray = functions.grayscale(image)

blur_gray = functions.gaussian_blur(gray, 5)

canny_blur = functions.canny(blur_gray, 100, 200)

return canny_blur

process_on =0

count =0

video= cv2.VideoWriter("output.avi", cv2.VideoWriter_fourcc(*"MJPG"), 10, (1600, 1200))

vidcap = cv2.VideoCapture('input.mp4')

success,image = vidcap.read()

success = True

while success:

processed = process_image(image)

video.write(processed)

这是我得到的错误：

OpenCV错误：在cv :: mjpeg :: MotionJpegWriter :: write中，断言失败(img.cols == width && img.rows == height * 3)，文件D：\ Build \ OpenCV \ opencv-3.2.0 \ modules \ videoio \ src \ cap_mjpeg_encoder.cpp，第834行回溯(最近一次调用最后一次)：文件“W：/Roborace/03_Information/10_Data/01_Montreal/camera/right_front_camera/01_Processed/Roborace_CAMERA_create_video.py”，第30行，在video.write中(已处理)cv2.error：D：\ Build \ OpenCV \ opencv-3.2.0 \ modules \ videoio \ src \ cap_mjpeg_encoder.cpp：834：错误：(-215)img.cols == width && img.rows ==函数cv :: mjpeg :: MotionJpegWriter :: write中的高度* 3

我的建议是：由于RGB色场，正常图像有3个维度 . 处理后的图像只有一个维度 . 我怎样才能在cv2.VideoWriter函数中调整它 .

谢谢你的帮助