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

test max faces arg

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2024-07-21 08:57:45 +01:00
parent a34daa9c83
commit 37b8cc7d31
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_represent.py
View File

@ -7,6 +7,7 @@ from deepface.commons import logger as log
logger = log.get_singletonish_logger()
def test_standard_represent():
img_path = "dataset/img1.jpg"
embedding_objs = DeepFace.represent(img_path)
@ -48,3 +49,14 @@ def test_represent_for_skipped_detector_backend_with_preloaded_image():
assert "h" in img_obj["facial_area"].keys()
assert "face_confidence" in img_obj.keys()
logger.info("✅ test represent function for skipped detector and preloaded image done")
def test_max_faces():
# confirm that input image has more than one face
results = DeepFace.represent(img_path="dataset/couple.jpg")
assert len(results) > 1
# test it with max faces arg
max_faces = 1
results = DeepFace.represent(img_path="dataset/couple.jpg", max_faces=max_faces)
assert len(results) == max_faces