github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 03:55:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update test_extract_faces.py

Browse Source 
type controls in detected face
This commit is contained in:
Sefik Ilkin Serengil 2024-08-30 08:04:50 +01:00 committed by GitHub
parent 5e261e709c
commit 0d1bce3618
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests/test_extract_faces.py
View File

@ -36,8 +36,26 @@ def test_different_detectors():
assert "right_eye" in img_obj["facial_area"].keys() assert "right_eye" in img_obj["facial_area"].keys()
right_eye = img_obj["facial_area"]["right_eye"] right_eye = img_obj["facial_area"]["right_eye"]
left_eye = img_obj["facial_area"]["left_eye"] left_eye = img_obj["facial_area"]["left_eye"]
# left eye and right eye must be tuple
assert isinstance(left_eye, tuple)
assert isinstance(right_eye, tuple)
# right eye should be the right eye of the person
assert left_eye[0] > right_eye[0] assert left_eye[0] > right_eye[0]
# left eye and right eye must be int not to have problem in api
assert isinstance(left_eye[0], int)
assert isinstance(left_eye[1], int)
assert isinstance(right_eye[0], int)
assert isinstance(right_eye[1], int)
# confidence must be float, not numpy not to have problem in api
assert "confidence" in img_obj.keys() assert "confidence" in img_obj.keys()
type_conf = type(img_obj["confidence"])
assert isinstance(
img_obj["confidence"], float
), f"confidence type must be float but it is {type_conf}"
# we added black pixeled borders to image because if faces are close to border, # we added black pixeled borders to image because if faces are close to border,
# then alignment moves them to outside of the image. adding this borders may # then alignment moves them to outside of the image. adding this borders may