Update app/app_utils.py
Browse files- app/app_utils.py +4 -2
app/app_utils.py
CHANGED
|
@@ -57,8 +57,10 @@ def preprocess_image_and_predict(inp):
|
|
| 57 |
cur_face_hm = np.float32(cur_face_hm) / 255
|
| 58 |
heatmap = show_cam_on_image(cur_face_hm, grayscale_cam, use_rgb=True)
|
| 59 |
|
| 60 |
-
|
| 61 |
-
|
|
|
|
|
|
|
| 62 |
|
| 63 |
def preprocess_video_and_predict(video):
|
| 64 |
|
|
|
|
| 57 |
cur_face_hm = np.float32(cur_face_hm) / 255
|
| 58 |
heatmap = show_cam_on_image(cur_face_hm, grayscale_cam, use_rgb=True)
|
| 59 |
|
| 60 |
+
return cur_face, heatmap, confidences
|
| 61 |
+
|
| 62 |
+
else:
|
| 63 |
+
return None, None
|
| 64 |
|
| 65 |
def preprocess_video_and_predict(video):
|
| 66 |
|