1. 程式人生 > >OpenCV-Python[第二章]——視訊初步

OpenCV-Python[第二章]——視訊初步

簡介

本小節學習讀取/顯示/儲存視訊。涉及到的主要函式是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函式對幀進行翻轉,對翻轉後的視訊幀進行輸出儲存以及顯示。最後是進行資源的釋放和視窗的銷燬。