import cv2
import time
import os
cap = cv2.VideoCapture('./video10.mp4') #導(dǎo)入視頻，可以將視頻放入和程序所在的同一目錄下，也可以放置別的目錄，修改對應(yīng)的路徑即可，我所用的是將視頻文件放置當(dāng)前目錄下的情況。
# saving = False
saving=True # 控制視頻是否將視頻逐幀保存為圖片
frame_id = 0
dir1 = ""
dir1 = time.strftime("%Y_%m_%d_%H_%M_%S", time.localtime())
os.mkdir(dir1)
saving = True
frame_id = 0
while True:
    # try:
    _, im = cap.read()
    if im is None:
        break
    cv2.imshow('name', im)
    key = cv2.waitKey(10) & 0xFF
    if saving:
        file_name = dir1 + "\\" + str(frame_id)
        cv2.imwrite(file_name + ".jpg", im)
        frame_id += 1
    if (key == ord('q')) | (key == 27):
        break
    # 如果需要按鍵控制開始視頻保存為圖像的時機(jī)，可以使用下面的代碼。
    # if key == ord('s') or key == ord('S'):
    #     if not saving:
    #         dir1 = time.strftime("%Y_%m_%d_%H_%M_%S", time.localtime())
    #         os.mkdir(dir1)
    #         saving = True
    #         frame_id = 0
    #     else:
    #         saving = False
    #
    # if saving:
    #     file_name = dir1 + "\\" + str(frame_id)
    #     cv2.imwrite(file_name + ".jpg", im)
    #
    #     frame_id += 1
    # if (key == ord('q')) | (key == 27):
    #     break