1. 程式人生 > 程式設計 >Python OpenCV視訊擷取並儲存實現程式碼

Python OpenCV視訊擷取並儲存實現程式碼

這篇文章主要介紹了Python OpenCV視訊擷取並儲存實現程式碼,文中通過示例程式碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下

在影象處理之前,我們需要對拿到手的資料進行篩選,對於視訊,我們需要從中擷取我們需要的一段或幾段

整體思路比較簡單,通過設定擷取視訊的起止時間(幀數),可以將該時間段內的影象儲存為新的視訊

直接上程式碼

"""
【函式名稱】 擷取視訊 
【引數】 輸入引數 視訊檔名稱
【詳細介紹】 輸入不同時間段 進行擷取拼接
【建立日期】 20191128 by wangxioabei
【修改日期】 NOTE.1: 
"""
def CutVideoFromFile(video_file_name,windows_name = 'videoShowing'):
  cap = cv2.VideoCapture(video_file_name) # 開啟視訊檔案
  # 需要明確視訊儲存的格式
  fourcc = cv2.VideoWriter_fourcc(*'XVID')
  fps = cap.get(cv2.CAP_PROP_FPS)
  print('幀率:%d'%fps)
  size = (int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)),int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)))
  out = cv2.VideoWriter('BeltTear.avi',fourcc,fps,size)
  SaveTime = [[38*60+38,39*60+59],[42*60+54,44*60+11],[47*60+8,48*60+24],[51*60+20,52*60+39]]
  print(SaveTime[0][0])
  now_frame = 0
  while (cap.isOpened()):
    ret,frame = cap.read() # 捕獲一幀影象
    img_h,img_w,img_ch = frame.shape
    # print(frame.shape)
    if ret:
      # 【1】 不能直接將灰度或二值化的圖片儲存成視訊,需要轉換成彩色
      if img_ch==1:
        frame = cv2.cvtColor(frame,cv2.COLOR_GRAY2BGR)
      # cv2.imshow(windows_name,frame)
      for i in range(len(SaveTime)):
        if now_frame > SaveTime[i][0]*fps and now_frame < SaveTime[i][1]*fps:
          out.write(frame)
          print(now_frame)
      now_frame += 1
      if now_frame > SaveTime[2][1]*fps:
        break;
      k = cv2.waitKey(1) & 0xFF
      if k == 27:
        break
      # cv2.waitKey(25)
    else:
      break
  cap.release()
  out.release()
  cv2.destroyAllWindows()

其中SaveTime列表 中巢狀的列表為擷取視訊的時間s,也可以將其當做引數傳入,即可封裝整個函式。

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。