in source/frontend/detector.py [0:0]
def __init__(self, env):
try:
self.FACE_AREA_THRESHOLD = env['FaceAreaThreshold']
self.FACE_MARGIN_RATIO = env['FaceMarginRatio']
self.USE_DEEP_LEANING_FOR_DETECTOR = env['UseDeepLeaningForDetector']
except KeyError:
print('Invalid config file')
raise
if self.USE_DEEP_LEANING_FOR_DETECTOR:
print('Loading SSD model...')
self.net = cv2.dnn.readNetFromCaffe(self.PROTOTEXT, self.CAFFEMODEL)
else:
print('Loading Haar-cascade classifier...')
self.face_cascade = cv2.CascadeClassifier(self.HAARLIKE)
print('Loaded!')