利用OpenCV進行H264視頻編碼的簡易方式


在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編碼,需要從源碼編譯,並在編譯過程中添加對應的支持庫。看到一篇帖子有介紹,目前還沒有驗證。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM