Created
April 24, 2019 21:06
-
-
Save yerkbn/b78b7d09f0f963938f2860cd09877a10 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
while True: | |
# Grab a single frame of video | |
ret, frame = video_capture.read() | |
# Resize frame of video to 1/4 size for faster face recognition processing | |
small_frame = cv2.resize(frame, (0, 0), fx=QUALITY, fy=QUALITY) | |
# Convert the image from BGR color (which OpenCV uses) to RGB color (which face_recognition uses) | |
rgb_small_frame = small_frame[:, :, ::-1] | |
# Only process every other frame of video to save time | |
if process_this_frame: | |
# Find all the faces and face encodings in the current frame of video | |
print('###', rgb_small_frame) | |
face_locations = face_recognition.face_locations(rgb_small_frame) | |
face_encodings = face_recognition.face_encodings(rgb_small_frame, face_locations) | |
face_names = [] | |
for face_encoding in face_encodings: | |
# See if the face is a match for the known face(s) | |
matches = face_recognition.compare_faces(known_face_encodings, face_encoding) | |
name = "Unknown" | |
# Or instead, use the known face with the smallest distance to the new face | |
face_distances = face_recognition.face_distance(known_face_encodings, face_encoding) | |
best_match_index = np.argmin(face_distances) | |
if matches[best_match_index]: | |
name = known_face_names[best_match_index] | |
face_names.append(name) | |
#process_this_frame = not process_this_frame | |
# print('####', name, face_locations) | |
# Display the results | |
for (top, right, bottom, left), name in zip(face_locations, face_names): | |
# Scale back up face locations since the frame we detected in was scaled to 1/4 size | |
top *= 4 | |
right *= 4 | |
bottom *= 4 | |
left *= 4 | |
# For saving | |
for family_name in CLASSES: | |
print(name) | |
if name in CLASSES[family_name]: | |
folder_name = family_name | |
cropped = frame[top-OFFSET_TOP:bottom+OFFSET, left-OFFSET:right+OFFSET] | |
image_path = "got/{}/{}_{}_{}.png".format(family_name, COUNT, name, 'vid2') | |
print(image_path) | |
COUNT+=1 | |
cv2.imwrite(image_path, cropped) | |
break | |
# Draw a box around the face | |
cv2.rectangle(frame, (left, top), (right, bottom), (0, 0, 255), 2) | |
# Draw a label with a name below the face | |
cv2.rectangle(frame, (left, bottom - 35), (right, bottom), (0, 0, 255), cv2.FILLED) | |
font = cv2.FONT_HERSHEY_DUPLEX | |
cv2.putText(frame, name, (left + 6, bottom - 6), font, 1.0, (255, 255, 255), 1) | |
# Display the resulting image | |
cv2.imshow('Video', frame) | |
# Hit 'q' on the keyboard to quit! | |
if cv2.waitKey(1) & 0xFF == ord('q'): | |
break | |
# Release handle to the webcam | |
video_capture.release() | |
cv2.destroyAllWindows() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment