從攝像頭獲取視頻
OpenCV提供了非常簡(jiǎn)單的接口可用于讀取攝像頭中的視頻流誊抛。要獲取視頻,首先需要使用cv.VideoCapture()創(chuàng)建一個(gè)VideoCapture對(duì)象醇坝。其創(chuàng)建參數(shù)可以是設(shè)備號(hào)、視頻文件名伏社、圖片序列饺藤、視頻流的URL,細(xì)節(jié)可看文檔干旧。
使用設(shè)備號(hào)創(chuàng)建對(duì)象時(shí)渠欺,傳入0即可打開(kāi)系統(tǒng)中默認(rèn)攝像頭。之后就可以使用cv.VideoCapture.read()方法按幀來(lái)獲取視頻圖片椎眯。使用cv.imshow()顯示圖片瓶埋。操作結(jié)束后需要使用cv.VideoCapture.release()方法釋放該VideoCapture對(duì)象怕品。
import numpy as np
import cv2 as cv
# 創(chuàng)建VideoCapture,傳入0即打開(kāi)系統(tǒng)默認(rèn)攝像頭
cap = cv.VideoCapture(0)
while(True):
# 讀取一幀,read()方法是其他兩個(gè)類方法的結(jié)合装悲,具體文檔
# ret為bool類型尤泽,指示是否成功讀取這一幀
ret, frame = cap.read()
# 就是個(gè)處理一幀的例子乖阵,這里轉(zhuǎn)為灰度圖
gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)
# 不斷顯示一幀栖疑,就成視頻了
# 這里沒(méi)有提前創(chuàng)建窗口,所以默認(rèn)創(chuàng)建的窗口不可調(diào)整大小
# 可提前使用cv.WINDOW_NORMAL標(biāo)簽創(chuàng)建個(gè)窗口
cv.imshow('frame',gray)
# 若沒(méi)有按下q鍵汞斧,則每1毫秒顯示一幀
if cv.waitKey(1) & 0xFF == ord('q'):
break
# 所有操作結(jié)束后不要忘記釋放
cap.release()
cv.destroyAllWindows()
如果終端不斷輸出“opencv Camera dropped frame!”夜郁,就說(shuō)明讀取幀的頻率過(guò)快,增大cv.waitKey()的傳入值就好了粘勒。
有的時(shí)候竞端,VideoCapture對(duì)象可能沒(méi)有被成功初始化,這時(shí)操作該對(duì)象會(huì)發(fā)生錯(cuò)誤庙睡∈赂唬可以使用cv.VideoCapture.isOpened()方法來(lái)判斷該攝像頭是否被打開(kāi)。如果沒(méi)有打開(kāi)乘陪,可以使用cv.VideoCapture.open()方法來(lái)打開(kāi)攝像頭赵颅。
可以使用cv.VideoCapture.get()方法來(lái)獲取已打來(lái)攝像頭的屬性。對(duì)應(yīng)的可以使用cv.VideoCapture.set()方法來(lái)設(shè)置這些屬性的值暂刘。比如,對(duì)于上面打開(kāi)的攝像頭捂刺,可以使用cap.get(cv.CAP_PROP_FRAME_WIDTH)和cap.get(cv.CAP_PROP_FRAME_HEIGHT)來(lái)分別獲取幀的寬和高谣拣。使用ret = cap.set(cv.CAP_PROP_FRAME_WIDTH,320)和ret = cap.set(cv.CAP_PROP_FRAME_HEIGHT,240)將幀的寬和高分別設(shè)為320和240。
操作視頻文件
使用視頻文件名創(chuàng)建VideoCapture對(duì)象族展,即可打開(kāi)對(duì)應(yīng)的視頻文件森缠。在顯示幀時(shí),注意使用cv.waitKey()來(lái)控制幀率仪缸,通常25毫秒是可以的贵涵。示例代碼如下:
import numpy as np
import cv2 as cv
cap = cv.VideoCapture('vtest.avi')
while(cap.isOpened()):
ret, frame = cap.read()
gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)
cv.imshow('frame',gray)
if cv.waitKey(1) & 0xFF == ord('q'):
break
cap.release()
cv.destroyAllWindows()
我打開(kāi)了一個(gè)mp4文件來(lái)顯示,但是效果并不好,弄得電腦很卡宾茂,不知道為什么瓷马。沒(méi)有攝像頭的流暢。
保存視頻
處理了視頻的每一幀之后跨晴,想要將其保存下來(lái)欧聘。對(duì)于圖片使用cv.imwrite()就可以了,但對(duì)于視頻端盆,多需要一些操作怀骤。
首先需要使用cv.VideoWriter()創(chuàng)建一個(gè)VideoWriter對(duì)象,創(chuàng)建對(duì)象時(shí)需要指定輸出文件名焕妙、FourCC編碼蒋伦、fps和幀的大小,最后一個(gè)參數(shù)為isColor焚鹊,其指定了是彩色幀(True)還是灰度幀(False)痕届,默認(rèn)為True。
FourCC是一個(gè)四字節(jié)碼寺旺,用于指定視頻的編碼方式爷抓。全部可用的FourCC在這里:fourcc.org,它是平臺(tái)相關(guān)的阻塑。
對(duì)于MJPG蓝撇,可以使用cv.VideoWriter_fourcc('M','J','P','G')或cv.VideoWriter_fourcc(*'MJPG')來(lái)設(shè)置FourCC。下面代碼從攝像頭中采集圖片陈莽,左右翻轉(zhuǎn)后存入視頻文件:
import numpy as np
import cv2 as cv
cv.namedWindow('frame', cv.WINDOW_NORMAL)
cap = cv.VideoCapture(0)
# 這里如果不設(shè)置窗口大小的話
# 存儲(chǔ)視頻文件時(shí)會(huì)出錯(cuò)
cap.set(cv.CAP_PROP_FRAME_WIDTH, 640)
cap.set(cv.CAP_PROP_FRAME_HEIGHT, 480)
fourcc = cv.VideoWriter_fourcc(*'MJPG')
out = cv.VideoWriter('output.mp4',fourcc, 20.0, (640,480))
while cap.isOpened():
ret, frame = cap.read()
if ret==True:
frame = cv.flip(frame,1)
out.write(frame)
cv.imshow('frame',frame)
if cv.waitKey(1) & 0xFF == ord('q'):
break
else:
break
cap.release()
out.release()
cv.destroyAllWindows()
雖然運(yùn)行上面代碼可以得到一個(gè)mp4文件渤昌,但是用播放器播放時(shí)確說(shuō)無(wú)法打開(kāi),不知道是為什么走搁。