Update utils/emotion_detection_brainai.py
Browse files
utils/emotion_detection_brainai.py
CHANGED
|
@@ -39,8 +39,8 @@ class EmotionModel:
|
|
| 39 |
|
| 40 |
return input_img
|
| 41 |
|
| 42 |
-
def detect_faces(uploaded_img_cv,
|
| 43 |
-
input_img = self.preprocess(
|
| 44 |
result_face = self.face_compiled_model([input_img])[self.face_output_layer]
|
| 45 |
boxes = self.post_process_face(result_face, uploaded_img_cv, conf)
|
| 46 |
|
|
|
|
| 39 |
|
| 40 |
return input_img
|
| 41 |
|
| 42 |
+
def detect_faces(uploaded_img_cv, conf = 0.5):
|
| 43 |
+
input_img = self.preprocess(uploaded_img_cv, self.face_input_layer)
|
| 44 |
result_face = self.face_compiled_model([input_img])[self.face_output_layer]
|
| 45 |
boxes = self.post_process_face(result_face, uploaded_img_cv, conf)
|
| 46 |
|