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

enhance batched detector test

Browse Source
This commit is contained in:
galthran-wq 2025-02-12 16:05:16 +00:00
parent 1bd83356e7
commit bbf6a55037
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/test_extract_faces.py
View File

@ -80,7 +80,7 @@ def test_different_detectors():
@pytest.mark.parametrize("detector_backend", [
# "YOLO",
"yolov11n",
"opencv",
])
def test_batch_extract_faces(detector_backend):
@ -89,8 +89,19 @@ def test_batch_extract_faces(detector_backend):
"dataset/img3.jpg",
"dataset/img11.jpg",
]
img_objs = DeepFace.extract_faces(img_path=img_paths, detector_backend=detector_backend)
assert len(img_objs) == 3
# Extract faces one by one
img_objs_individual = [DeepFace.extract_faces(img_path=img_path, detector_backend=detector_backend)[0] for img_path in img_paths]
# Extract faces in batch
img_objs_batch = DeepFace.extract_faces(img_path=img_paths, detector_backend=detector_backend)
assert len(img_objs_batch) == len(img_objs_individual)
for img_obj_individual, img_obj_batch in zip(img_objs_individual, img_objs_batch):
assert np.array_equal(img_obj_individual["face"], img_obj_batch["face"])
assert img_obj_individual["facial_area"] == img_obj_batch["facial_area"]
assert img_obj_individual["confidence"] == img_obj_batch["confidence"]
def test_backends_for_enforced_detection_with_non_facial_inputs():