Building a Real-Time Face Recognition System Using Python, OpenCV, and face_recognition

1pip install opencv-python face-recognition numpy
2

1import face_recognition
2import cv2
3import numpy as np
4
5# Initialize the webcam
6video_capture = cv2.VideoCapture(0)
7
8# Load and encode known faces
9kosal_image = face_recognition.load_image_file("kosal.jpg")
10kosal_face_encoding = face_recognition.face_encodings(kosal_image)[0]
11
12lika_image = face_recognition.load_image_file("lika.jpg")
13lika_face_encoding = face_recognition.face_encodings(lika_image)[0]
14
15# Store the encodings and corresponding names in arrays
16known_face_encodings = [
17    kosal_face_encoding,
18    lika_face_encoding,
19]
20known_face_names = [
21    "Kosal",
22    "Molika",
23]
24
25# Initialize variables for face recognition
26face_locations = []
27face_encodings = []
28face_names = []
29process_this_frame = True
30
31while True:
32    # Capture a frame from the webcam
33    ret, frame = video_capture.read()
34
35    if not ret:
36        print("Failed to grab frame")
37        break
38
39    # Process every other frame for performance
40    if process_this_frame:
41        # Resize the frame to 1/4 for faster processing
42        small_frame = cv2.resize(frame, (0, 0), fx=0.25, fy=0.25)
43
44        # Convert BGR image to RGB
45        rgb_small_frame = cv2.cvtColor(small_frame, cv2.COLOR_BGR2RGB)
46
47        # Detect faces and get encodings
48        face_locations = face_recognition.face_locations(rgb_small_frame)
49        face_encodings = face_recognition.face_encodings(rgb_small_frame, face_locations)
50
51        face_names = []
52        for face_encoding in face_encodings:
53            # Check if the face matches any known faces
54            matches = face_recognition.compare_faces(known_face_encodings, face_encoding)
55            name = "Unknown"
56
57            # Find the closest match
58            face_distances = face_recognition.face_distance(known_face_encodings, face_encoding)
59            best_match_index = np.argmin(face_distances)
60            if matches[best_match_index]:
61                name = known_face_names[best_match_index]
62
63            face_names.append(name)
64
65    process_this_frame = not process_this_frame
66
67    # Draw boxes around faces and label them
68    for (top, right, bottom, left), name in zip(face_locations, face_names):
69        # Scale the coordinates back to the original size
70        top *= 4
71        right *= 4
72        bottom *= 4
73        left *= 4
74
75        # Draw a box around the face
76        cv2.rectangle(frame, (left, top), (right, bottom), (0, 0, 255), 2)
77
78        # Label the face with a name
79        cv2.rectangle(frame, (left, bottom - 35), (right, bottom), (0, 0, 255), cv2.FILLED)
80        font = cv2.FONT_HERSHEY_DUPLEX
81        cv2.putText(frame, name, (left + 6, bottom - 6), font, 1.0, (255, 255, 255), 1)
82
83    # Display the video feed
84    cv2.imshow('Video', frame)
85
86    # Break the loop on 'q' key press
87    if cv2.waitKey(1) & 0xFF == ord('q'):
88        break
89
90# Release the webcam and close windows
91video_capture.release()
92cv2.destroyAllWindows()
93