File size: 1,402 Bytes
c4ba2f9 9b36524 c4ba2f9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | from src.geometry import FaceGeometry
def extract_features(landmarks, hairline_y=None):
geo = FaceGeometry(landmarks, hairline_y=hairline_y)
thirds = geo.facial_thirds_ratio()
if thirds:
u = thirds["upper"]
m = thirds["middle"]
l = thirds["lower"]
max_dev = max(abs(u - 0.333), abs(m - 0.333), abs(l - 0.333))
if max_dev < 0.05:
thirds_balance = 0.0
elif max_dev < 0.10:
thirds_balance = 0.5
else:
thirds_balance = 1.0
else:
u = m = l = 0.33
thirds_balance = 0.0
return {
"face_ratio": geo.face_ratio(),
"jaw_ratio": geo.jaw_ratio(),
"jaw_to_height": geo.jaw_to_height(),
"eye_ratio": geo.eye_ratio(),
"eye_height": geo.eye_height(),
"lip_ratio": geo.lip_ratio(),
"nose_position": geo.nose_position_ratio(),
"lower_face_ratio": geo.lower_face_ratio(),
"chin_prominence": geo.chin_prominence(),
"symmetry": geo.symmetry_score(),
"upper_third": thirds["upper"] if thirds else 0.33,
"middle_third": thirds["middle"] if thirds else 0.33,
"lower_third": thirds["lower"] if thirds else 0.33,
"upper_third": u,
"middle_third": m,
"lower_third": l,
"mid_lower_ratio": geo.middle_lower_ratio(),
"thirds_balance": thirds_balance,
} |