我正在训练一个模型,该模型将获取给定视频的每一帧,并返回用于使用seaborn生成热图的值,并收集该热图并将其保存在新视频中
我成功获得了热图框架,但是我无法将其直接保存在新视频中,我想执行以下代码,但这是行不通的
video = cv2.VideoWriter(video_path, 0, 10, (width, height))
for heat_map in heat_maps:
heat_map = sns.heatmap(heat_map, cmap = 'jet', vmin=0, vmax=160)
video.write(heat_map)
video.release()
这会出现以下错误
TypeError: Expected Ptr<cv::UMat> for argument 'image'
有一种方法可以将每个热图保存为图像,然后再将其写入视频,但我想知道是否还有一种更直接的方法来将海上热图保存到视频中