1. 程式人生 > 程式設計 >python使用openCV遍歷資料夾裡所有視訊檔案並儲存成圖片

python使用openCV遍歷資料夾裡所有視訊檔案並儲存成圖片

如果你在資料夾裡有很多視訊,並且資料夾裡還有資料夾,資料夾裡的資料夾也有視訊,怎麼能逐個讀取並且儲存。。所以我寫了個程式碼用了os,walk,這個可以遍歷所有資料夾裡的檔案和資料夾

import os
import cv2
cut_frame = 250 # 多少幀截一次,自己設定就行
save_path = "C:\文獻與資料\手持紅外\圖片"
for root,dirs,files in os.walk(r"C:\文獻與資料\手持紅外"): # 這裡就填資料夾目錄就可以了
 for file in files:
 # 獲取檔案路徑
 if ('.mp4' in file):
  path = os.path.join(root,file)
  video = cv2.VideoCapture(path)
  video_fps = int(video.get(cv2.CAP_PROP_FPS))
  print(video_fps)
  current_frame = 0
  while (True):
  ret,image = video.read()
  current_frame = current_frame + 1
  if ret is False:
   video.release()
   break
  if current_frame % cut_frame == 0:
   # cv2.imwrite(save_path + '/' + file[:-4] + str(current_frame) + '.jpg',#  image) # file[:-4]是去掉了".mp4"字尾名,這裡我的命名格式是,視訊檔名+當前幀數+.jpg,使用imwrite就不能有中文路徑和中文檔名
   cv2.imencode('.jpg',image)[1].tofile(save_path + '/' + file[:-4] + str(current_frame) + '.jpg') #使用imencode就可以整個路徑中可以包括中文,檔名也可以是中文
   print('正在儲存' + file + save_path + '/' + file[:-4] + str(current_frame))

ps:下面看下python 遍歷資料夾

import os
# 遍歷資料夾
def walkFile(file):
 for root,files in os.walk(file):
 # root 表示當前正在訪問的資料夾路徑
 # dirs 表示該資料夾下的子目錄名list
 # files 表示該資料夾下的檔案list
 # 遍歷檔案
 for f in files:
  print(os.path.join(root,f))
 # 遍歷所有的資料夾
 for d in dirs:
  print(os.path.join(root,d))
def main():
 walkFile("f:/ostest/")
if __name__ == '__main__':
 main()

總結

以上所述是小編給大家介紹的python使用openCV遍歷資料夾裡所有視訊檔案並儲存成圖片,希望對大家有所幫助,如果大家有任何疑問請給我留言,小編會及時回覆大家的。在此也非常感謝大家對我們網站的支援!
如果你覺得本文對你有幫助,歡迎轉載,煩請註明出處,謝謝!