diff options
Diffstat (limited to 'face-detection/01_face_capture_dataset.py')
| -rw-r--r-- | face-detection/01_face_capture_dataset.py | 17 |
1 files changed, 12 insertions, 5 deletions
diff --git a/face-detection/01_face_capture_dataset.py b/face-detection/01_face_capture_dataset.py index edf271e..8c1883e 100644 --- a/face-detection/01_face_capture_dataset.py +++ b/face-detection/01_face_capture_dataset.py @@ -42,7 +42,7 @@ while True: scaleFactor=1.1,# how much the image size is reduced at each image scale-10% reduction minNeighbors=5, # how many neighbors each candidate rectangle should have to retain it minSize=(30, 30)# Minimum possible object size. Objects smaller than this size are ignored. - ) + ) for (x,y,w,h) in faces: # Create a bounding box across the detected face cv2.rectangle(frame, (x,y), (x+w,y+h), BOXCOLOR, 3) # 5 parameters - frame, topleftcoords,bottomrightcooords,boxcolor,thickness @@ -51,8 +51,15 @@ while True: # if dataset folder doesnt exist create: if not os.path.exists("dataset"): os.makedirs("dataset") - # Save the captured bounded-grayscaleimage into the datasets folder - cv2.imwrite("dataset/User." + str(face_id) + '.' + str(count) + ".jpg", frameGray[y:y+h,x:x+w]) + # Save the captured bounded-grayscaleimage into the datasets folder only if the same file doesn't exist + file_path = os.path.join("dataset", f"User.{face_id}.{count}.jpg") + if os.path.exists(file_path): + # Move the existing file to the "old_dataset" folder + old_file_path = file_path.replace("dataset", "old_dataset") + os.rename(file_path, old_file_path) + # Write the newer images after moving the old images + cv2.imwrite(file_path, frame_gray[y:y+h, x:x+w]) + # Display the original frame to the user cv2.imshow('FaceCapture', frame) @@ -63,8 +70,8 @@ while True: break elif key == 113: # q key break - elif count >= COUNT_LIMIT: # Take COUNT_LIMIT face sample and stop video capture - break + elif count >= COUNT_LIMIT: # Take COUNT_LIMIT face samples and stop video capture + break # Release the camera and close all windows print("\n [INFO] Exiting Program and cleaning up stuff") |