fix: retinaface too slow
Browse files
app.py
CHANGED
@@ -83,7 +83,7 @@ def predict_and_compare(image):
|
|
83 |
distance = result["distance"]
|
84 |
|
85 |
# Get face area for SSIM
|
86 |
-
face_area = DeepFace.extract_faces(temp_path
|
87 |
|
88 |
# Calculate SSIM against SSIM reference
|
89 |
ssim_value = compare_face_ssim(image, ssim_reference, face_area)
|
|
|
83 |
distance = result["distance"]
|
84 |
|
85 |
# Get face area for SSIM
|
86 |
+
face_area = DeepFace.extract_faces(temp_path)[0]["facial_area"]
|
87 |
|
88 |
# Calculate SSIM against SSIM reference
|
89 |
ssim_value = compare_face_ssim(image, ssim_reference, face_area)
|