使用Python，OpenCV捕獲關(guān)鍵事件，并進(jìn)行視頻剪輯

• • 1. 效果圖
  • 2. 原理
  • • 2.1 步驟
    • 2.2 如何找到HSV空間值呢？
• [[[60 255 255]]]
• • 3. 源碼
  • • 3.1 keyclipwriter.py
    • 3.2 save_key_events.py
  • 參考

上一篇博客介紹了如何使用Python、OpenCV寫入視頻文件，這一篇將介紹如何做有趣的視頻剪輯，而不是保留整個視頻文件；

總體目標(biāo)是構(gòu)建視頻概要，將視頻流的最關(guān)鍵，突出和有趣部分提煉成一系列短視頻文件。

使用場景有：

• 訪問在區(qū)域中檢測到的運(yùn)動物體；
• 截取入侵者進(jìn)入房子或公寓的片段；

什么才是視頻的關(guān)鍵幀、有趣部分，則完全取決于您的需求和目標(biāo)；

使用此方法，可以將小時級的視頻流素材縮小為多個秒級的有趣事件的小視頻，有效地產(chǎn)生視頻概要。

1. 效果圖

檢測畫面中的綠色盆，并將其生成gif圖，如下：

2. 原理

2.1 步驟

1. 定義關(guān)鍵事件；
2. 將包含事件的視頻片段寫入到視頻文件；

利用線程以確保在輸入流和輸出視頻剪輯文件中執(zhí)行I / O時，不會放慢主程序的速度。
利用內(nèi)置的Python數(shù)據(jù)結(jié)構(gòu)，如DEQUE隊(duì)列以按順序存放關(guān)鍵幀；

2.2 如何找到HSV空間值呢？

比如要尋找綠色，可以找到綠色對應(yīng)的BGR(0,255,0) ,利用opencv轉(zhuǎn)換為HSV，然后取[ H-10,S,V]為下限 – [ H+10,S,V]為上限；

green = np.uint8([[[0, 255, 0]]])
hsv_green = cv2.cvtColor(green, cv2.COLOR_BGR2HSV)
print(hsv_green)

[[[60 255 255]]]

3. 源碼

3.1 keyclipwriter.py

# 導(dǎo)入必要的包
from collections import deque # 導(dǎo)入隊(duì)列，FIFO（first in first out）先進(jìn)先出原則
from threading import Thread
from queue import Queue
import time
import cv2class KeyClipWriter:# bufSize 要在內(nèi)存緩沖區(qū)中緩存的最大幀數(shù)# timeout 超時時間def __init__(self, bufSize=64, timeout=1.0):# 存儲在內(nèi)存中要保留的最大緩沖區(qū)大小 以及睡眠超時期間self.bufSize = bufSizeself.timeout = timeout# 初始化幀的緩沖區(qū)（64幀）、Q：待寫入的視頻幀、視頻寫入類、 多線程（以避免I/O延遲）# recordering boolean類型指示錄制是否已啟動self.frames = deque(maxlen=bufSize)self.Q = Noneself.writer = Noneself.thread = Noneself.recording = Falsedef update(self, frame):# 更新幀緩沖區(qū)self.frames.appendleft(frame)# 如果正在記錄，將幀加入隊(duì)列if self.recording:self.Q.put(frame)def start(self, outputPath, fourcc, fps):# 展示正在記錄關(guān)鍵事件幀# 表明正在錄制，啟動視頻寫入類，初始化需要寫入視頻文件的幀隊(duì)列self.recording = Trueself.writer = cv2.VideoWriter(outputPath, fourcc, fps,(600,600), True)self.Q = Queue()# 循環(huán)遍歷所有幀，并加入隊(duì)列for i in range(len(self.frames), 0, -1):self.Q.put(self.frames[i - 1])# 開啟一個單獨(dú)的線程寫入幀到視頻文件self.thread = Thread(target=self.write, args=())self.thread.daemon = Trueself.thread.start()def write(self):# 保持循環(huán)while True:# 如果停止了錄制，退出線程if not self.recording:return# 判斷隊(duì)列是否已經(jīng)沒有關(guān)鍵事件幀了if not self.Q.empty():# 獲取隊(duì)列中的下一幀并寫入視頻文件frame = self.Q.get()self.writer.write(frame)# 如果隊(duì)列已經(jīng)空了，sleep超時時間以不浪費(fèi)cpu周期# 使用隊(duì)列時空閑一會兒尤其重要，隊(duì)列數(shù)據(jù)結(jié)構(gòu)是線程安全的，暗示必須在更新內(nèi)部緩沖區(qū)之前獲取鎖/信號量。# 如果不sleep，當(dāng)緩沖區(qū)為空時，然后寫入和更新方法將不斷為鎖而戰(zhàn)斗。相反，最好讓視頻寫入類等待一會兒，直到需要寫入文件的隊(duì)列中存在積壓的幀。else:time.sleep(self.timeout)def flush(self):# 通過刷新所有剩余幀來清空隊(duì)列while not self.Q.empty():frame = self.Q.get()self.writer.write(frame)def finish(self):# 表明結(jié)束錄制、加入線程# 刷新隊(duì)列中的所有幀到視頻文件# 釋放視頻寫入指針self.recording = Falseself.thread.join()self.flush()self.writer.release()

3.2 save_key_events.py

# USAGE
# python save_key_events.py --output output#  導(dǎo)入必要的包
from pyimagesearch.keyclipwriter import KeyClipWriter
from imutils.video import VideoStream
import argparse
import datetime
import imutils
import time
import cv2# 構(gòu)建命令行參數(shù)及解析
# --output 輸出視頻剪輯文件的路徑及名稱
# --picamera 是否使用樹莓?dāng)z像頭，默認(rèn)-1：不是
# --fps 輸出視頻剪輯的幀率（即每秒的幀數(shù)） 默認(rèn)20
# --codec 輸出視頻剪輯的四維編解碼器 默認(rèn)MJPG
# --buffer-size 視頻剪切類的緩沖區(qū)大小 默認(rèn)32，用于存儲來自相機(jī)傳感器最近輪詢幀的內(nèi)存緩沖區(qū)的大小。
#               較大的緩沖大小將允許在“關(guān)鍵事件”中包含在輸出視頻剪輯中的“關(guān)鍵事件”之前和之后的更多上下文，而較小的緩沖大小將在“關(guān)鍵事件”之前和之后存儲更少的幀；
ap = argparse.ArgumentParser()
ap.add_argument("-o", "--output", required=True,help="path to output directory")
ap.add_argument("-p", "--picamera", type=int, default=-1,help="whether or not the Raspberry Pi camera should be used")
ap.add_argument("-f", "--fps", type=int, default=20,help="FPS of output video")
ap.add_argument("-c", "--codec", type=str, default="MJPG",help="codec of output video")
ap.add_argument("-b", "--buffer-size", type=int, default=32,help="buffer size of video clip writer")
args = vars(ap.parse_args())# 初始化視頻流，并預(yù)熱相機(jī)傳感器2s
print("[INFO] warming up camera...")
vs = VideoStream(usePiCamera=args["picamera"] > 0).start()
time.sleep(2.0)# 定義綠色球的較低和上邊界HSV顏色空間
greenLower = (29, 86, 6)
greenUpper = (64, 255, 255)# 初始化關(guān)鍵事件剪輯類，
# 初始化用于計(jì)算尚未包含任何有趣事件的連續(xù)幀數(shù)的整數(shù)。
kcw = KeyClipWriter(bufSize=args["buffer_size"])
consecFrames = 0# 保持循環(huán)
while True:# 獲取當(dāng)前幀，縮放，初始化一個布爾類型值updateConsecFrames表示是否連續(xù)幀計(jì)數(shù)器應(yīng)該被更新frame = vs.read()frame = imutils.resize(frame, width=600)updateConsecFrames = True# 高斯模糊幀，并將其RGB顏色空間轉(zhuǎn)換為HSV顏色空間，因此可以使用顏色閾值blurred = cv2.GaussianBlur(frame, (11, 11), 0)hsv = cv2.cvtColor(blurred, cv2.COLOR_BGR2HSV)# 為綠色構(gòu)建一個顏色mask，然后執(zhí)行一系列腐蝕膨脹操作已移除mask中小斑點(diǎn)# cv2.inRange實(shí)際進(jìn)行顏色閾值處理，cv2.erode腐蝕，cv2.dilate膨脹操作mask = cv2.inRange(hsv, greenLower, greenUpper)mask = cv2.erode(mask, None, iterations=2)mask = cv2.dilate(mask, None, iterations=2)# 在mask中查找輪廓cnts = cv2.findContours(mask.copy(), cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)cnts = imutils.grab_contours(cnts)# 當(dāng)發(fā)現(xiàn)輪廓時繼續(xù)處理if len(cnts) > 0:# 尋找mask中最大的輪廓，計(jì)算最小外接圓c = max(cnts, key=cv2.contourArea)((x, y), radius) = cv2.minEnclosingCircle(c)updateConsecFrames = radius <= 10# 只有當(dāng)外接圓輪廓達(dá)到一定值時處理if radius > 10:# 重置連續(xù)幀計(jì)數(shù)器為0，繪制外接圓輪廓到幀上consecFrames = 0cv2.circle(frame, (int(x), int(y)), int(radius),(0, 0, 255), 2)# 如果還沒開始計(jì)關(guān)鍵事件幀數(shù)，開始記錄if not kcw.recording:timestamp = datetime.datetime.now()p = "{}/{}.avi".format(args["output"],timestamp.strftime("%Y%m%d-%H%M%S"))kcw.start(p, cv2.VideoWriter_fourcc(*args["codec"]),args["fps"])# 如果幀中沒有任何事件，將沒有關(guān)鍵事件幀的幀計(jì)數(shù)if updateConsecFrames:consecFrames += 1# 更新關(guān)鍵事件幀緩存區(qū)kcw.update(frame)# 如果沒有關(guān)鍵事件的幀達(dá)到一定數(shù)量，計(jì)數(shù)記錄剪輯if kcw.recording and consecFrames == args["buffer_size"]:kcw.finish()# 展示幀cv2.imshow("Frame", frame)key = cv2.waitKey(1) & 0xFF# 按下‘q’鍵，退出循環(huán)if key == ord("q"):break# 如果還在錄制中，結(jié)束錄制
if kcw.recording:kcw.finish()# 清理工作，釋放窗口，釋放視頻文件寫入指針
cv2.destroyAllWindows()
vs.stop()

參考

• https://www.pyimagesearch.com/2016/02/29/saving-key-event-video-clips-with-opencv/

總結(jié)

以上是生活随笔為你收集整理的使用Python，OpenCV捕获关键事件，并进行视频剪辑的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。