Сохраняйте тепловые карты seaborn прямо в видео с помощью opencv videowriter - PullRequest
0 голосов
/ 13 июля 2020

Я обучаю модель, которая затем берет каждый кадр данного видео, возвращает значения, которые используются для создания тепловых карт с использованием 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'

Есть способ сохранить все тепловые карты в виде изображений, а затем записать их на видео позже, но мне было интересно, есть ли более прямой способ сохранить тепловые карты морского дна в видео

...