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

batch test add other detector models

Browse Source
This commit is contained in:
galthran-wq 2025-02-18 10:13:03 +00:00
parent 70b61a7f4f
commit 27dea80248
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_extract_faces.py
View File

@ -84,15 +84,20 @@ def test_different_detectors():
"ssd",
"mtcnn",
"retinaface",
"yunet",
"centerface",
# optional
# "yolov11s",
# "mediapipe",
# "dlib",
])
def test_batch_extract_faces(detector_backend):
detector_backend_to_rtol = {
"opencv": 0.1,
"ssd": 0.01,
"mtcnn": 0.2,
"retinaface": 0.01,
"yolov11s": 0.03,
}
rtol = detector_backend_to_rtol[detector_backend]
rtol = detector_backend_to_rtol.get(detector_backend, 0.01)
img_paths = [
"dataset/img2.jpg",
"dataset/img3.jpg",