Update face_detection.py

face_detection.py

@@ -2,10 +2,11 @@ import insightface
 import numpy as np
 from PIL import Image, ImageOps
 import cv2
+from scipy.ndimage import gaussian_filter
 
 # Load InsightFace models
 try:
-    detector = insightface.model_zoo.get_model('retinaface_r50_v1')
+    detector = insightface.model_zoo.get_model('retinaface_mnet025_v2')
     detector.prepare(ctx_id=-1)  # Use CPU, set ctx_id=0 for GPU
 except Exception as e:
     print(f"Error loading face detection model: {e}")
@@ -52,8 +53,6 @@ def get_landmarks(image):
         landmarks = landmark_model.get(img, bbox)
         yield landmarks
 
-# Remaining functions remain unchanged
-
 def image_align(src_img, face_landmarks, output_size=512, transform_size=2048, enable_padding=True, x_scale=1, y_scale=1, em_scale=0.1, alpha=False):
     # Align function modified from ffhq-dataset
     # See https://github.com/NVlabs/ffhq-dataset for license