使用Python+OpenCV将多级嵌套文件夹下的视频文件抽帧为JPG图片

一、使用Python+OpenCV将多级嵌套文件夹下的视频文件抽帧为JPG图片

2. import os
3. import cv2
4. import time
6. # 存放视频文件的多层嵌套文件夹路径
7. videoPath = 'D:\\videos\\'
8. # 保存抽帧的图片的文件夹路径
9. savePath = 'D:\\images\\'
11. if not os.path.exists(savePath):
12.     os.mkdir(savePath)
13. video_num = 0
15. for root, dirs, files in os.walk(videoPath):
16.     # 提取多级嵌套文件夹路径下的文件列表
17.     for video in files:
18.         print(video)
19.         if(video.endswith(".avi")):
20.             c = 1
21.             videofile = os.path.join(root, video)
22.             vc = cv2.VideoCapture(videofile) #读入视频文件
23.             if vc.isOpened(): #判断是否正常打开
24.                 video_num = video_num + 1
25.                 rval , frame = vc.read()
26.                 #print("正常打开：",video)
27.                 if(video_num%100 == 0):
28.                     print(video_num)
29.             else:
30.                 rval = False
31.                 print("打开失败：",video)
32.                 continue
33.    
34.             timeF = 25  #视频帧计数间隔频率
35.             try:
36.                 while rval:   #循环读取视频帧
37.                     rval, frame = vc.read()
38.                     if(c%timeF == 0): #每隔timeF帧进行存储操作
39.                         cv2.imwrite(os.path.join(savePath, video[:-4] + '_' + str(time.time_ns()) + '.jpg'),frame) #存储为图像
40.                     c = c + 1
41.                     cv2.waitKey(1)
42.             except:
43.                 #print("error")
44.                 continue
45.             vc.release()

复制代码

此中：timeF变量是界说视频抽帧隔断频率，可修改。
videoPath修改为本身的视频地点目次，savePath修改为本身的图片生存的目次。