-
-
Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathmain_recognize_image.py
37 lines (30 loc) · 1.79 KB
/
main_recognize_image.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# main_recognize_image.py
# Usage: %python main_recognize_image.py
# Import the FaceDetect class
from FaceDetect.facedetect import FaceDetect
# Initialize FaceDetect
# Params:
# - settings (optional): Dictionary with settings to be passed to the FaceDetector
# * mode: image or video (default)
# * custom: False (default). If you wish to extend the FaceDetect class, specify the method that it needs to execute
# * method: call native callback methods during detection or bypass with a custom method
# * draw: draws the detection on the canvas if set to True (default)
# * print: prints the face locations and labels on the console
# * face-extraction: extracts captures of the faces into their own images. Applicable only to mode image
# * face-features: Draws the specified face features. Off by default. Pass the list ['face'] to draw the whole face
# * known-faces: Setting need for facial recognition when 'method' is set to 'recognize'
# It is a dictionary of face labels and image paths associated.
# For example: {'John': 'person1.png', 'Jane': 'person2.png'}
#
# Set the method to recognize and provide a dictionary of known faces names and image path
facedetector = FaceDetect({'mode': 'image', 'method': 'recognize', 'known-faces': {'John': 'resources/person1.png',
'Jane': 'resources/person2.png'}})
try:
# When the start method is not given an image or video path, it starts the webcam
# For Image file: facedetector.start('<path to image file>')
# For Video: facedetector.start('<path to video file>')
# Press 'q' to exit
facedetector.start('resources/people.jpg')
# FaceDetect always generates a FaceDetect Exception
except Exception as error:
print(error)