-
-
Notifications
You must be signed in to change notification settings - Fork 5
/
main_detection_webcam_extended.py
43 lines (33 loc) · 1.77 KB
/
main_detection_webcam_extended.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
38
39
40
41
42
43
# main_detection_webcam.py
# Usage: %python main_detection_webcam_extended.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'}
#
class MyDetector(FaceDetect):
# example of a method that print the detections property from FaceDetect
def main(self):
print(self.detections)
# Passing settings (the default ones are being passed for comparison purposes with main.py)
facedetector = MyDetector({'custom': True, 'method': 'main'})
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()
# FaceDetect always generates a FaceDetect Exception
except Exception as error:
print(error)