From 4577b430515919f53ff510b2fdc2abd5ed5e8cfe Mon Sep 17 00:00:00 2001 From: Sefik Ilkin Serengil Date: Fri, 12 Apr 2024 14:49:07 +0100 Subject: [PATCH] unit test extended for centerface --- tests/visual-test.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/tests/visual-test.py b/tests/visual-test.py index 09b4b8d..71ddcb6 100644 --- a/tests/visual-test.py +++ b/tests/visual-test.py @@ -34,9 +34,10 @@ detector_backends = [ "retinaface", "yunet", "yolov8", + "centerface", ] - +""" # verification for model_name in model_names: obj = DeepFace.verify( @@ -59,7 +60,7 @@ dfs = DeepFace.find( ) for df in dfs: logger.info(df) - +""" expand_areas = [0] img_paths = ["dataset/img11.jpg", "dataset/img11_reflection.jpg"] @@ -75,7 +76,7 @@ for expand_area in expand_areas: ) for face_obj in face_objs: face = face_obj["face"] - logger.info(detector_backend) + logger.info(f"testing {img_path} with {detector_backend}") logger.info(face_obj["facial_area"]) logger.info(face_obj["confidence"]) @@ -99,7 +100,10 @@ for expand_area in expand_areas: le_x = face_obj["facial_area"]["left_eye"][0] assert re_x < le_x, "right eye must be the right eye of the person" - assert isinstance(face_obj["confidence"], float) + type_conf = type(face_obj["confidence"]) + assert isinstance( + face_obj["confidence"], float + ), f"confidence type must be float but it is {type_conf}" assert face_obj["confidence"] <= 1 plt.imshow(face)