OpenCV-Python[第二章]——視訊初步
阿新 • • 發佈:2019-02-15
簡介
本小節學習讀取/顯示/儲存視訊。涉及到的主要函式是cv2.VideoCapture(),cv2.VideoWriter().
捕捉視訊
使用cv2.VideoCapture即可捕捉視訊,它的引數可以是數字或者檔名,並且可以返回一個VideoCapture物件。如果是數字則代表連線的攝像頭,有多個攝像頭的情況依次為0,1,2.檔名則表示從本地視訊檔案獲取視訊流。之後就可以一幀幀捕捉視訊並逐幀處理、顯示等。最後需要釋放,否則攝像頭會一直處於開啟狀態。
儲存視訊
使用cv2.VideoWriter()建立一個VideoWriter物件,其用法如下:
# Define the codec and create VideoWriter object
fourcc = cv2.VideoWriter_fourcc(*'X264')
out = cv2.VideoWriter('output.avi',fourcc, 20.0, (640,480))
引數依次為輸出檔名、FourCC、幀率、大小。其中FourCC是一個4位元組的程式碼用於指定所用編碼器。關於FourCC的更多細節可以檢視fourcc.org常見的FourCC碼有:DIVX,XVID,MJPG,X264,WMV1,WMV2等。
示例程式碼
import numpy as np
import cv2
cap = cv2.VideoCapture(0)
#cap = cv2.VideoCapture('test.avi') #use local file to create capture
# Define the codec and create VideoWriter object
fourcc = cv2.VideoWriter_fourcc(*'X264')
out = cv2.VideoWriter('output.avi',fourcc, 25.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()
程式碼很簡單,開啟攝像頭建立一個捕捉視訊流,以及建立一個輸出視訊流儲存視訊檔案。while迴圈中逐幀讀取視訊幀,使用cv2.flip函式對幀進行翻轉,對翻轉後的視訊幀進行輸出儲存以及顯示。最後是進行資源的釋放和視窗的銷燬。