lern-tracking-system/raspberry-pi/sensors/camera_calibration-master/data_generation.py

'''This script is for generating data
1. Provide desired path to store images.
2. Press 'c' to capture image and display it.
3. Press any button to continue.
4. Press 'q' to quit.
'''

import cv2

camera = cv2.VideoCapture(0)
ret, img = camera.read()


#path = "/home/abhishek/stuff/object_detection/explore/aruco_data/" # !!
count = 0
while True:
    name = path + str(count)+".jpg"
    ret, img = camera.read()
    cv2.imshow("img", img)


    if cv2.waitKey(20) & 0xFF == ord('c'):
        cv2.imwrite(name, img)
        cv2.imshow("img", img)
        count += 1
        if cv2.waitKey(0) & 0xFF == ord('q'):

            break;