switching to yolo8

.deepface/weights/yolov8n-face.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d545bf1add5aa736a4febac4f4f9245a6d596cd0fe70d5d57989fe0cb9e626ca
+size 6389512

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 *.db filter=lfs diff=lfs merge=lfs -text
 face.json filter=lfs diff=lfs merge=lfs -text
+.deepface/weights/yolov8n-face.pt filter=lfs diff=lfs merge=lfs -text

app.py

@@ -8,15 +8,9 @@ import pyzipper
 import numpy as np
 import gradio as gr
 from annoy import AnnoyIndex
-from deepface.commons import functions
-from deepface.basemodels import Facenet512
+from deepface import DeepFace
 
 
-# load the face model
-model = Facenet512.loadModel()
-
-input_shape_x, input_shape_y = functions.find_input_shape(model)
-
 index = AnnoyIndex(512, "euclidean")
 index.load(f"face.db")
 
@@ -27,7 +21,6 @@ with pyzipper.AESZipFile('persons.zip') as zf:
     zf.setpassword(password)
     PERFORMER_DB = json.loads(zf.read('performers.json'))
 
-
 ## Prediction functions
 
 
@@ -42,17 +35,7 @@ def image_search_performer(image, threshold=20.0, results=3):
     """
     
     image_array = np.array(image)
-
-    img = functions.preprocess_face(
-        img=image_array,
-        target_size=(input_shape_x, input_shape_y),
-        enforce_detection=True,
-        detector_backend="retinaface",
-        align=True,
-    )
-
-    img = functions.normalize_input(img, normalization="Facenet2018")
-    face = model.predict(img)[0]
+    face = DeepFace.represent(img_path = image_array, detector_backend='yolov8', model_name='Facenet512')[0]['embedding']
     return search_performer(face, threshold, results)
 
 

face.db

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:dcd67774910249b57b6ec70dcda2a7acf9ce5e6675a7090f2978be2a23b78943
-size 618376464
+oid sha256:7ab68d919e7a875bcfdf5ab4e4d60d76cbd744cdba8ab255274f28d578be5483
+size 612925440

face.json

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6828967799e1074fa321d445b4644a1769e37cf980dff499c5ab3ea5690df0d8
-size 7708760
+oid sha256:30606fb842f9ebdb2a1018dab31b949b7f785b45ce30a7625e724aa267c03d34
+size 7679120

persons.zip

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ce27efb585af9ee61762bb4c90ff686bee4745e715b4a3fb9e980d869db012b8
+oid sha256:adf87972c704d73affed311b663d24f5fc790d3f06b35ae9c734d0babd06d212
 size 3705286

requirements.txt

@@ -15,7 +15,8 @@ charset-normalizer==3.1.0
 click==8.1.3
 contourpy==1.0.7
 cycler==0.11.0
-deepface==0.0.75
+git+https://github.com/serengil/deepface.git#f13aa34feb17ccd539eb67b671a42d7160d59298
+ultralytics==8.0.175
 entrypoints==0.4
 fastapi==0.95.0
 ffmpy==0.3.0
@@ -53,7 +54,7 @@ mdit-py-plugins==0.3.3
 mdurl==0.1.2
 mtcnn==0.1.1
 multidict==6.0.4
-numpy==1.21.5
+numpy
 oauthlib==3.2.2
 opencv-python==4.7.0.72
 opt-einsum==3.3.0