一、問題

博主最近在做深度學習的目標檢測問題，需要用到自制的圖片資料集。但是用攝像頭拍的話，太慢了。於是博主就拍了一些視訊，然後按幀將它儲存下來。

二、開發環境

python3

OpenCV

windows10

三、程式碼實現

import cv2
import os

videos_src_path = r'./G5'#提取圖片的視訊資料夾
videos_save_path = r'./G5'#儲存圖片的路徑

videos = os.listdir(videos_src_path) #用於返回指定的資料夾包含的檔案或資料夾的名字的列表。
videos = filter(lambda x: x.endswith('mp4'), videos)#將mp4檔案讀進來，可改為avi等格式

for each_video in videos:
    frame_count = 1
    # 得到每個資料夾的名字, 並指定每一幀的儲存路徑
    each_video_name, _ = each_video.split('.')
    os.mkdir(videos_save_path + '/' + each_video_name)
    each_video_save_full_path = os.path.join(videos_save_path, each_video_name) + '/'

    # 得到完整的視訊路徑
    each_video_full_path = os.path.join(videos_src_path, each_video)
    # 用OpenCV一幀一幀讀取出來
    cap = cv2.VideoCapture(each_video_full_path)
    
    success = True
    while (success):
        success, frame = cap.read()
        print('Read a new frame: ', success)
        params = []
        params.append(1)
        #if len(frame) > 0:
        #if success:
        if frame is not None and frame_count%3==0:#每？幀取一幀圖片儲存下來，可以自己修改
            cv2.imwrite(each_video_save_full_path + each_video_name + "_%d.jpg" % frame_count, frame, params)
            print(frame_count)
        frame_count = frame_count + 1
    cap.release()