Update utils/emotion_detection_brainai.py
Browse files
utils/emotion_detection_brainai.py
CHANGED
|
@@ -40,7 +40,7 @@ class EmotionModel:
|
|
| 40 |
return input_img
|
| 41 |
|
| 42 |
def detect_faces(self, img, conf = 0.5):
|
| 43 |
-
input_img = self.preprocess( img,
|
| 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 |
|
|
|
|
| 40 |
return input_img
|
| 41 |
|
| 42 |
def detect_faces(self, img, conf = 0.5):
|
| 43 |
+
input_img = self.preprocess( img, 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 |
|