diff --git a/tests/visual-test.py b/tests/visual-test.py new file mode 100644 index 0000000..dd95451 --- /dev/null +++ b/tests/visual-test.py @@ -0,0 +1,50 @@ +import matplotlib.pyplot as plt +from deepface import DeepFace + +model_names = [ + "VGG-Face", + "Facenet", + "Facenet512", + "OpenFace", + "DeepFace", + "DeepID", + "Dlib", + "ArcFace", + "SFace", +] +detector_backends = ["opencv", "ssd", "dlib", "mtcnn", "retinaface"] + +# verification +for model_name in model_names: + obj = DeepFace.verify( + img1_path="dataset/img1.jpg", img2_path="dataset/img2.jpg", model_name=model_name + ) + print(obj) + print("---------------------") + +# represent +for model_name in model_names: + embedding_objs = DeepFace.represent(img_path="dataset/img1.jpg", model_name=model_name) + for embedding_obj in embedding_objs: + embedding = embedding_obj["embedding"] + print(f"{model_name} produced {len(embedding)}D vector") + +# find +dfs = DeepFace.find( + img_path="dataset/img1.jpg", db_path="dataset", model_name="Facenet", detector_backend="mtcnn" +) +for df in dfs: + print(df) + +# extract faces +for detector_backend in detector_backends: + face_objs = DeepFace.extract_faces( + img_path="dataset/img1.jpg", detector_backend=detector_backend + ) + for face_obj in face_objs: + face = face_obj["face"] + print(detector_backend) + plt.imshow(face) + plt.axis("off") + plt.show() + print("-----------")