From 2fd9f3702185e3dff99d39b55889d8fbb250d9ac Mon Sep 17 00:00:00 2001
From: Markus Hinsche <m.hinsche@gmail.com>
Date: Mon, 11 Nov 2024 14:35:30 +0100
Subject: [PATCH] Choose lightweight model GhostFaceNet

---
 scripts/mh_download_models.py   | 56 +++++++++++++++++++++++++++++++++
 scripts/mh_use_api_represent.py |  2 +-
 scripts/mh_use_api_verify.py    |  2 +-
 3 files changed, 58 insertions(+), 2 deletions(-)
 create mode 100644 scripts/mh_download_models.py

diff --git a/scripts/mh_download_models.py b/scripts/mh_download_models.py
new file mode 100644
index 0000000..c2b73b3
--- /dev/null
+++ b/scripts/mh_download_models.py
@@ -0,0 +1,56 @@
+from pathlib import Path
+import requests
+import json
+
+URL_VERIFY = "http://127.0.0.1:5005/verify"
+HEADERS = {"Content-Type": "application/json"}
+
+models = [
+    # "VGG-Face",
+    # "Facenet",
+    # "Facenet512",
+    # "OpenFace",
+    # "DeepFace",
+    # "DeepID",
+    # "ArcFace",
+    "Dlib",
+    # "SFace",
+    # "GhostFaceNet",
+]
+
+if __name__ == "__main__":
+    result = {
+        "results": [
+            {
+                "embedding": [
+                    -1.2134509086608887,
+                    -1.1512534618377686,
+                ],
+                "face_confidence": 0.92,
+                "facial_area": {
+                    "h": 998,
+                    "left_eye": [1292, 1300],
+                    "right_eye": [912, 1283],
+                    "w": 998,
+                    "x": 621,
+                    "y": 906,
+                },
+            }
+        ]
+    }
+
+    vec1 = result["results"][0]["embedding"]
+    vec2 = vec1
+
+    for model in models:
+        payload = {
+            "model_name": model,
+            "img1_path": vec1,
+            "img2_path": vec2,
+        }
+
+        response = requests.request(
+            "POST", URL_VERIFY, data=json.dumps(payload), headers=HEADERS
+        )
+
+        print(response.text)
diff --git a/scripts/mh_use_api_represent.py b/scripts/mh_use_api_represent.py
index 99184dd..3f176b5 100644
--- a/scripts/mh_use_api_represent.py
+++ b/scripts/mh_use_api_represent.py
@@ -22,7 +22,7 @@ if __name__ == "__main__":
     )
     b64encoded_string = get_base64_from_file(image_path)
 
-    payload = {"model_name": "Facenet", "img": b64encoded_string}
+    payload = {"model_name": "GhostFaceNet", "img": b64encoded_string}
 
     # img_path = "https://raw.githubusercontent.com/serengil/deepface/refs/heads/master/tests/dataset/couple.jpg"
     # payload = {"model_name": "Facenet", "img": img_path}
diff --git a/scripts/mh_use_api_verify.py b/scripts/mh_use_api_verify.py
index 1281444..53a1e97 100644
--- a/scripts/mh_use_api_verify.py
+++ b/scripts/mh_use_api_verify.py
@@ -156,7 +156,7 @@ if __name__ == "__main__":
     vec2 = vec1
 
     payload = {
-        "model_name": "Facenet",
+        "model_name": "GhostFaceNet",
         "img1_path": vec1,
         "img2_path": vec2,
     }