spuun commited on
Commit
5862051
·
verified ·
1 Parent(s): 21b013f

fix: retinaface too slow

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, detector_backend='retinaface')[0]["facial_area"]
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)