python讀取和保存視頻文件

為了獲取視頻，應(yīng)該創(chuàng)建一個(gè) VideoCapture 對(duì)象。他的參數(shù)可以是設(shè)備的索引號(hào)，或者是一個(gè)視頻文件。設(shè)備索引號(hào)就是在指定要使用的攝像頭。

一般的筆記本電腦都有內(nèi)置攝像頭。所以參數(shù)就是 0。你可以通過設(shè)置成 1 或者其他的來選擇別的攝像頭。之后，你就可以一幀一幀的捕獲視頻了。但是最后，別忘了停止捕獲視頻。

import numpy as np 
import cv2 
cap=cv2.VideoCapture(0) 
while (True): 
  ret,frame=cap.read() 
  gray=cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY) 
  cv2.imshow("shiyan",gray) 
  if cv2.waitKey(1)&0xFF==ord('q'): 
    break 
cap.release() 
cv2.destroyAllWindows() 

與c++很類似

cap.read() 返回一個(gè)布爾值（True/False）。如果幀讀取的是正確的，就是 True。

所以最后你可以通過檢查他的返回值來查看視頻文件是否已經(jīng)到了結(jié)尾。
有時(shí) cap 可能不能成功的初始化攝像頭設(shè)備。這種情況下上面的代碼會(huì)報(bào)錯(cuò)。你可以使用 cap.isOpened()，來檢查是否成功初始化了。如果返回值是True，那就沒有問題。否則就要使用函數(shù) cap.open()。

 你可以使用函數(shù) cap.get(propId) 來獲得視頻的一些參數(shù)信息。這里propId 可以是 0 到 18 之間的任何整數(shù)。每一個(gè)數(shù)代表視頻的一個(gè)屬性其中的一些值可以使用 cap.set(propId,value) 來修改， value 就是你想要設(shè)置成的新值。例如，我可以使用 cap.get(3) 和 cap.get(4) 來查看每一幀的寬和高。默認(rèn)情況下得到的值是 640X480。但是我可以使用 ret=cap.set(3,320)和 ret=cap.set(4,240) 來把寬和高改成 320X240。

2、保存視頻

在我們捕獲視頻，并對(duì)每一幀都進(jìn)行加工之后我們想要保存這個(gè)視頻。對(duì)于圖片來時(shí)很簡(jiǎn)單只需要使用 cv2.imwrite()。但對(duì)于視頻來說就要多做點(diǎn)工作。

   這次我們要?jiǎng)?chuàng)建一個(gè) VideoWriter 的對(duì)象。我們應(yīng)該確定一個(gè)輸出文件的名字。接下來指定 FourCC 編碼。播放頻率和幀的大小也都需要確定。最后一個(gè)是 isColor 標(biāo)簽。如果是 True，每一幀就是彩色圖，否則就是灰度圖。FourCC 就是一個(gè) 4 字節(jié)碼，用來確定視頻的編碼格式?？捎玫木幋a列表可以從fourcc.org查到。這是平臺(tái)依賴的。下面這些編碼器對(duì)我來說是有用個(gè)。

• In Fedora: DIVX, XVID, MJPG, X264, WMV1, WMV2. (XVID is
more preferable. MJPG results in high size video. X264 gives
very small size video)
• In Windows: DIVX (More to be tested and added)
• In OSX : (I don't have access to OSX. Can some one fill this?)

FourCC 碼以下面的格式傳給程序，以 MJPG 為例：

cv2.cv.FOURCC('M','J','P','G') 或者 cv2.cv.FOURCC(*'MJPG')。

下面的代碼是從攝像頭中捕獲視頻，沿水平方向旋轉(zhuǎn)每一幀并保存它。

import numpy as np 
import cv2 
cap = cv2.VideoCapture(0) 
# Define the codec and create VideoWriter object 
fourcc = cv2.cv.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) 
  out.write(frame) 
  cv2.imshow("實(shí)驗(yàn)",frame) 
  if cv2.waitKey(0)&0xFF==ord('q'): 
   break 
 else: 
  break 
cap.release() 
out.release() 
cv2.destroyAllWindows() 

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持億速云。