diff --git a/collect_imgs.py b/collect_imgs.py index 6a51698da809aecb8efe3e91582dcab1c795b806..dfaf01d592bf0353270bbd76501756dd48bdb977 100644 --- a/collect_imgs.py +++ b/collect_imgs.py @@ -7,15 +7,15 @@ DATA_DIR = './data' if not os.path.exists(DATA_DIR): os.makedirs(DATA_DIR) -number_of_classes = 3 -dataset_size = 100 +nombre_de_classe = 3 +taille_du_dataset= 100 cap = cv2.VideoCapture(0) -for j in range(number_of_classes): +for j in range(nombre_de_classe): if not os.path.exists(os.path.join(DATA_DIR, str(j))): os.makedirs(os.path.join(DATA_DIR, str(j))) - print('Collecting data for class {}'.format(j)) + print('Collecte image pour la classe {}'.format(j)) done = False while True: @@ -27,7 +27,7 @@ for j in range(number_of_classes): break counter = 0 - while counter < dataset_size: + while counter < taille_du_dataset: ret, frame = cap.read() cv2.imshow('frame', frame) cv2.waitKey(25)