mirror/EyePy
1
0
Fork 0
mirror of https://github.com/ck-zhang/EyePy.git synced 2026-01-01 09:09:37 -06:00
Code Issues Packages Projects Releases Wiki Activity

attempt at face check

Browse Source
This commit is contained in:
will wade
2024-09-29 07:59:25 +01:00
committed by ck-zhang
parent 3a93092211
commit e2ca2c720e
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
gaze_estimator.py
View File

@@ -204,6 +204,19 @@ def run_calibration(gaze_estimator, camera_index=0):
p0 = points[i]
p1 = points[i + 1]
# Loop until a valid face is detected
valid_face_detected = False
while not valid_face_detected:
ret, frame = cap.read()
if not ret:
continue
# Extract features from the current frame
features = gaze_estimator.extract_features(frame)
if features is not None:
valid_face_detected = True
print(f"Face detected for calibration point {p0}")
for frame_idx in range(N):
ret, frame = cap.read()
if not ret:
@@ -221,6 +234,7 @@ def run_calibration(gaze_estimator, camera_index=0):
cv2.imshow("Calibration", canvas)
cv2.waitKey(1)
# Extract features from the current frame
features = gaze_estimator.extract_features(frame)
if features is not None:
features_list.append(features)