如何根据用户操作有选择地切换OpenCV的VideoWriter

时间:2016-06-16 16:17:03

标签: python opencv

假设我有一些视频剪辑应该显示特定的步骤序列,但是当创建剪辑时,它们可能在事件之前或之后包含了不需要的操作。

是否可以使用OpenCV逐帧自动播放视频,以便操作员可以在看到所需操作的开始时点击一个键,然后在序列完成时点击一个键并保存视频的那部分作为新的更小更精确的序列视频。

  

下面的代码将逐帧读取网络摄像头,在写入视频之前翻转框架,直到用户按下键盘上的q键。

我如何确保用户在进入时可以观看帧流,然后当他们看到他们感兴趣的事件时,他们可以切换VideoWriter以开始将这些帧写入磁盘,但操作员不需要不断点击一个键将每个帧发送到VideoWriter,当操作员看到事件结束时,他们可以再次关闭VideoWriter

import numpy as np
import cv2

cap = cv2.VideoCapture(0)

# Define the codec and create VideoWriter object
fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('output.avi',fourcc, 20.0, (640,480))

while(cap.isOpened()):
    ret, frame = cap.read()
    if ret==True:
        frame = cv2.flip(frame,0)

        # write the flipped frame
        out.write(frame)

        cv2.imshow('frame',frame)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break
    else:
        break

# Release everything if job is finished
cap.release()
out.release()
cv2.destroyAllWindows()

1 个答案:

答案 0 :(得分:1)

在考虑之后,逐帧编写视频不会很好地利用资源,也可能无法扩展。

我将展开此代码以将事件数据写入CSV文件,并将其与MoviePy结合使用,以根据鼠标左键按下时记录的时间戳提取子剪辑

如果其他人可以改进解决方案,我欢迎他们的意见

import cv2
import numpy as np
cap = cv2.VideoCapture('YourVideoFile.mp4')

#Define the Mouse Callback Function
def record_action(event,x,y,flags,param):
    if event == cv2.EVENT_LBUTTONDOWN:
        print "Left Button Down @ " + str(cap.get(cv2.CAP_PROP_POS_MSEC)) + " milliseconds into video \n"
    elif event == cv2.EVENT_LBUTTONUP:
        print "Left Button Up @ " + str(cap.get(cv2.CAP_PROP_POS_MSEC)) + " milliseconds into video \n"

#Need to use a Named Window so it can be referenced in the mouse definition 
#and used when outputting the frames from the video in the imshow call later on
cv2.namedWindow("RecordMe")
#Bind the function above to the window
cv2.setMouseCallback("RecordMe",record_action)

while True:
    ret, frame = cap.read()
    #Use NamedWindow we created earlier to show the frames
    cv2.imshow('RecordMe',frame)
    if cv2.waitKey(30) & 0xff == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()