Spaces:
Running
Running
Add time measurement and exception handling to image_search_performers function
Browse files
app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
import os
|
2 |
import io
|
|
|
3 |
import json
|
4 |
import math
|
5 |
import base64
|
@@ -50,11 +51,16 @@ def image_search_performers(image, threshold=20.0, results=3):
|
|
50 |
image_array = np.array(image)
|
51 |
|
52 |
response = []
|
53 |
-
|
54 |
t = time.time()
|
55 |
-
|
56 |
-
|
57 |
-
|
|
|
|
|
|
|
|
|
|
|
58 |
print(f"Time to find faces: {time.time() - t}")
|
59 |
for face in faces:
|
60 |
embedding = face['embedding']
|
|
|
1 |
import os
|
2 |
import io
|
3 |
+
import time
|
4 |
import json
|
5 |
import math
|
6 |
import base64
|
|
|
51 |
image_array = np.array(image)
|
52 |
|
53 |
response = []
|
54 |
+
|
55 |
t = time.time()
|
56 |
+
try:
|
57 |
+
faces = DeepFace.represent(img_path = image_array, detector_backend='retinaface', model_name='Facenet512', normalization="Facenet2018")
|
58 |
+
# faces = DeepFace.represent(img_path = image_array, detector_backend='yolov8', model_name='Facenet512', normalization="Facenet2018")
|
59 |
+
# faces = DeepFace.represent(img_path = image_array, detector_backend='mtcnn', model_name='Facenet512', normalization="Facenet2018")
|
60 |
+
except ValueError as e:
|
61 |
+
print(e)
|
62 |
+
raise gr.Error("No faces found in the image")
|
63 |
+
|
64 |
print(f"Time to find faces: {time.time() - t}")
|
65 |
for face in faces:
|
66 |
embedding = face['embedding']
|