2015-08-20 2 views

ответ

0

Вы можете получить кадры из видео с помощью OpenCV, так как в этом example, и построить график поверх с Matplotlib, например,

import cv2 
import matplotlib.pyplot as plt 
import numpy as np 

filename = './SampleVideo.mp4' 
cap = cv2.VideoCapture(filename) 

frames = int(cap.get(cv2.cv.CV_CAP_PROP_FRAME_COUNT)) 
width = int(cap.get(cv2.cv.CV_CAP_PROP_FRAME_WIDTH)) 
height = int(cap.get(cv2.cv.CV_CAP_PROP_FRAME_HEIGHT)) 

fig, ax = plt.subplots(1,1) 
plt.ion() 
plt.show() 

#Setup a dummy path 
x = np.linspace(0,width,frames) 
y = x/2. + 100*np.sin(2.*np.pi*x/1200) 

for i in range(frames): 
    fig.clf() 
    flag, frame = cap.read() 

    plt.imshow(frame) 
    plt.plot(x,y,'k-', lw=2) 
    plt.plot(x[i],y[i],'or') 

    if cv2.waitKey(1) == 27: 
     break 

У меня есть видео от here.

+0

Я получил ваш пример работы. Как я могу сохранить видеофайл с графиками, нарисованными на нем? – tylerjw

+0

Привет @tylerjw, вы можете использовать cv2 'VideoWriter' или' matplotlib' 'manimation' writer. Лично я хотел бы сохранить кадры в файл, а затем построить видео с изображений, используя 'savefig ('./ out {: 05d}'. Format (i) .png', чтобы получить ряд файлов с последующими именами, а затем собрать вместе с 'ffmpeg' или что-то ... –

+0

Спасибо. Я обнаружил, что matplotlib слишком медленный для моего приложения (не в режиме реального времени, но быстрее, чем несколько секунд/кадров), поэтому я попытаюсь написать его с помощью opencv и их инструменты рисования. – tylerjw

Смежные вопросы