怎么在python中使用opencv實(shí)現(xiàn)視頻抽幀

怎么在python中使用opencv實(shí)現(xiàn)視頻抽幀，很多新手對此不是很清楚，為了幫助大家解決這個難題，下面小編將為大家詳細(xì)講解，有這方面需求的人可以來學(xué)習(xí)下，希望你能有所收獲。

1、數(shù)據(jù)集簡述：

       雖然有主流龐大的COCO、VOC數(shù)據(jù)集，但是科研人員仍需要特殊領(lǐng)域要求的數(shù)據(jù)集，所以采用人工實(shí)地采集的方式進(jìn)行收集數(shù)據(jù)集圖像；通過拍照收集圖像過于繁瑣，所以通常是將攝像頭無規(guī)則的移動旋轉(zhuǎn)以及遠(yuǎn)近拉縮，進(jìn)而錄制視頻；再通過視頻抽幀的方式得到大量的圖像，再將這些圖像進(jìn)行人工標(biāo)注處理。抽幀后的圖像更具有泛化性。

2、代碼介紹：

下面是采用以幀數(shù)為間隔的方法進(jìn)行視頻抽幀，博主個人認(rèn)為這樣子的方式調(diào)節(jié)間隔更加方便，更符合個人習(xí)慣。

import cv2
from PIL import Image
import numpy as np


cap = cv2.VideoCapture("D:/Download/ANMR0005.mp4")  # 獲取視頻對象
isOpened = cap.isOpened  # 判斷是否打開
# 視頻信息獲取
fps = cap.get(cv2.CAP_PROP_FPS)

imageNum = 0
sum=0
timef=15  #隔15幀保存一張圖片

while (isOpened):

    sum+=1

    (frameState, frame) = cap.read()  # 記錄每幀及獲取狀態(tài)

    if frameState == True and (sum % timef==0):

        # 格式轉(zhuǎn)變，BGRtoRGB
        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
        # 轉(zhuǎn)變成Image
        frame = Image.fromarray(np.uint8(frame))

        frame = np.array(frame)

        # RGBtoBGR滿足opencv顯示格式
        frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)

        imageNum = imageNum + 1
        fileName = 'D:/Download/video_image/image' + str(imageNum) + '.jpg'  # 存儲路徑
        cv2.imwrite(fileName, frame, [cv2.IMWRITE_JPEG_QUALITY, 100])
        print(fileName + " successfully write in")  # 輸出存儲狀態(tài)

    elif frameState == False:
        break

print('finish!')
cap.release()

3、代碼效果：

pycharm運(yùn)行py文件后結(jié)果框顯示的內(nèi)容：

看完上述內(nèi)容是否對您有幫助呢？如果還想對相關(guān)知識有進(jìn)一步的了解或閱讀更多相關(guān)文章，請關(guān)注億速云行業(yè)資訊頻道，感謝您對億速云的支持。