在Python下,利用pip安裝預編譯的opencv庫,並實現h264格式的視頻編碼。
1. 安裝OpenCV
$ pip install opencv-python
建議在python虛擬環境下安裝,不容易產生相互影響。
2. 代碼示例
讀取筆記本自帶攝像頭,並保存為視頻的最簡實現。
import cv2
def main():
output_path = './output.avi'
vc = cv2.VideoCapture(0)
ret, frame = vc.read()
w = int(vc.get(cv2.CAP_PROP_FRAME_WIDTH))
h = int(vc.get(cv2.CAP_PROP_FRAME_HEIGHT))
fps = vc.get(cv2.CAP_PROP_FPS)
fourcc = cv2.VideoWriter_fourcc('H', '2', '6', '4')
# fourcc = cv2.VideoWriter_fourcc('X', 'V', 'I', 'D')
# fourcc = cv2.VideoWriter_fourcc('H', 'E', 'V', 'C')
vw = cv2.VideoWriter(output_path, fourcc, fps, (w, h), True)
while ret:
vw.write(frame)
ret, frame = vc.read()
cv2.imshow('frame', frame)
if cv2.waitKey(5) & 0xFF == ord('q'):
cv2.destroyAllWindows()
return -1
if __name__ == '__main__':
main()
2.1 默認編碼
預編譯的opencv,默認支持XVID格式的編碼,但是不支持H264。
如果將# fourcc = cv2.VideoWriter_fourcc('X', 'V', 'I', 'D')
注釋取消,程序可以正常運行並保存視頻。
2.2 H264支持
按示例代碼配置為H264編碼,會顯示如下錯誤提示。攝像頭可以采集到圖像,但是不能正常保存為視頻。
Failed to load OpenH264 library: openh264-1.8.0-win64.dll
Please check environment and/or download library: https://github.com/cisco/openh264/releases
[libopenh264 @ 000001e7a96ddec0] Incorrect library version loaded
Could not open codec 'libopenh264': Unspecified error
根據提示,到Github庫中搜索對應的動態庫。
注意選取對應的版本。我安裝的OpenCV版本為4.4.0.42
,提示需要的動態庫為openh264-1.8.0-win64.dll
,在Release頁中找到對應的版本,將下載獲得的dll文件解壓並放在與py文件同級的工程目錄下。
不同OpenCV版本可能需要的支持庫版本也不同,沒有測試,注意看提示就行。
如果考慮通用性也可以將dll所在的路徑加入環境變量Path中。
本示例實現是在Windows平台下,Linux下應該也是類似的操作,下載對應的h264支持庫文件,並加入到PATH能夠搜索到的地址。
重新運行程序,可以正常顯示圖像,並保存為視頻文件。
用Potplayer之類的播放器打開,可以在屬性中看到,編碼格式的相關信息:
視頻編碼: H264 - Native D3D9 DXVA Decoder(VLD)
2.3 H265支持
想着用同樣的方法嘗試一下H265編碼。
將# fourcc = cv2.VideoWriter_fourcc('H', 'E', 'V', 'C')
注釋取消,發現如下錯誤提示:
OpenCV: FFMPEG: tag 0x43564548/'HEVC' is not found (format 'avi / AVI (Audio Video Interleaved)')'
看起來目前預編譯版本的OpenCV是不支持H265編碼的。如果要用H265編碼,需要從源碼編譯,並在編譯過程中添加對應的支持庫。看到一篇帖子有介紹,目前還沒有驗證。