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

test batch extract faces

Browse Source
This commit is contained in:
galthran-wq 2025-02-12 15:47:24 +00:00
parent 0ad7c57abf
commit b38e95c407
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_extract_faces.py
View File

@ -79,6 +79,20 @@ def test_different_detectors():
logger.info(f"✅ extract_faces for {detector} backend test is done") logger.info(f"✅ extract_faces for {detector} backend test is done")
@pytest.mark.parametrize("detector_backend", [
# "YOLO",
"opencv",
])
def test_batch_extract_faces(detector_backend):
img_paths = [
"dataset/img2.jpg",
"dataset/img3.jpg",
"dataset/img11.jpg",
]
img_objs = DeepFace.extract_faces(img_path=img_paths, detector_backend=detector_backend)
assert len(img_objs) == 3
def test_backends_for_enforced_detection_with_non_facial_inputs(): def test_backends_for_enforced_detection_with_non_facial_inputs():
black_img = np.zeros([224, 224, 3]) black_img = np.zeros([224, 224, 3])
for detector in detectors: for detector in detectors: