vikhyatk commited on
Commit
3e0b66d
·
verified ·
1 Parent(s): 1161baa

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +1 -5
demo.py CHANGED
@@ -115,11 +115,7 @@ def process_image(input_image):
115
  gaze_points = []
116
 
117
  for face in faces:
118
- face_center = (
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