cap_images.py

import cv2
import os
import shutil
import argparse
from itertools import count

parser = argparse.ArgumentParser()
parser.add_argument(
    '-dir',
    '-directory',
    required=False,
    default='images',
    help='specify location of image folder'
)
args = parser.parse_args()
path = args.dir

vid = cv2.VideoCapture(0)
# WARNING: will empty the directory if it exists
if os.path.exists(path) and os.path.isdir(path):
    shutil.rmtree(path)
os.makedirs(path)

for index in count(0):
    ret, frame = vid.read()
    if not ret: break
    # cv2.namedWindow("output", cv2.WINDOW_NORMAL)  qqq
    cv2.imshow('output', frame)
    name = './{}/f-{}.jpg'.format(path, index)
    print("Saved {}\r".format(name), end='\r')
    cv2.imwrite(name, frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

vid.release()
cv2.destroyAllWindows()