使用opencv videowriter直接将Seaborn热图保存到视频中

时间:2020-07-13 10:17:05

标签: python opencv seaborn heatmap

我正在训练一个模型,该模型将获取给定视频的每一帧,并返回用于使用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'

有一种方法可以将每个热图保存为图像,然后再将其写入视频,但我想知道是否还有一种更直接的方法来将海上热图保存到视频中

0 个答案:

没有答案