import cv2
import os
i = 0
cap = cv2.VideoCapture(1, cv2.CAP_DSHOW)
while True:
# 获取图片帧
ret, frame = cap.read()
if ret == True:
cv2.imshow('img', frame)
k = cv2.waitKey(1)
# q退出,
if k == ord('q'):
cv2.destroyAllWindows()
break
# 输入空格保存图片
elif k == ord(' '):
i = i + 1
path = "./pic_capture"
folder = os.path.exists(path)
if not folder:
os.makedirs(path)
print("~~~~创建了文件夹~~~~")
cv2.imwrite(os.path.join(path, str(i) + '.jpg'), frame)
print("Frames{} Captured".format(i))