Premier commit

collect_imgs.py

@@ -10,7 +10,7 @@ if not os.path.exists(DATA_DIR):
 number_of_classes = 3
 dataset_size = 100
 
-cap = cv2.VideoCapture(2)
+cap = cv2.VideoCapture(0)
 for j in range(number_of_classes):
     if not os.path.exists(os.path.join(DATA_DIR, str(j))):
         os.makedirs(os.path.join(DATA_DIR, str(j)))

data/.gitignore

-.gitignore
-