Spaces:
Running
on
Zero
Running
on
Zero
Update demo.py
Browse files
demo.py
CHANGED
|
@@ -115,11 +115,7 @@ def process_image(input_image):
|
|
| 115 |
gaze_points = []
|
| 116 |
|
| 117 |
for face in faces:
|
| 118 |
-
|
| 119 |
-
(face["x_min"] + face["x_max"]) / 2,
|
| 120 |
-
(face["y_min"] + face["y_max"]) / 2,
|
| 121 |
-
)
|
| 122 |
-
gaze = model.detect_gaze(pil_image, face_center)["gaze"]
|
| 123 |
|
| 124 |
if gaze is None:
|
| 125 |
continue
|
|
|
|
| 115 |
gaze_points = []
|
| 116 |
|
| 117 |
for face in faces:
|
| 118 |
+
gaze = model.detect_gaze(pil_image, face=face)["gaze"]
|
|
|
|
|
|
|
|
|
|
|
|
|
| 119 |
|
| 120 |
if gaze is None:
|
| 121 |
continue
|