From baeb2249ee30144b3eb266d53117b6fae5dadb0c Mon Sep 17 00:00:00 2001 From: Saumit Dinesan Date: Tue, 9 May 2023 14:25:25 +0530 Subject: Enhanced code to handle creation of 'dataset' & 'trainer' folder for saving images and model --- face-detection/02_face_training.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'face-detection/02_face_training.py') diff --git a/face-detection/02_face_training.py b/face-detection/02_face_training.py index 68b21fe..a0d5a43 100644 --- a/face-detection/02_face_training.py +++ b/face-detection/02_face_training.py @@ -26,6 +26,9 @@ print ("\n [INFO] Training faces. It will take a few seconds. Wait ...") faces,ids = getImagesAndLabels(path) #Train the LBPH recognizer using the face samples and their corresponding labels recognizer.train(faces, np.array(ids)) +# if trainer folder doesnt exist create: +if not os.path.exists("trainer"): + os.makedirs("trainer") #save the model into trainer/trainer.yml recognizer.write('trainer/trainer.yml') # Print the numer of faces trained and then exit the program -- cgit v1.2.3