Fix an issue where faces are not correctly detected in images with high resolution when using RetinaFace detector

Author: dimitribarbot

liveportrait/utils/cropper_face_alignment.py

@@ -68,7 +68,7 @@ def __init__(self, **kwargs) -> None:
         if face_detector == 'blazeface':
             face_detector_kwargs = {'back_model': face_detector_dtype == 'blazeface_back_camera'}
         elif face_detector == 'retinaface':
-            face_detector_kwargs = {'fp16': face_detector == torch.float16}
+            face_detector_kwargs = {'fp16': face_detector == torch.float16, 'max_size': 1280}
         else:
             face_detector_kwargs = {}
 

liveportrait/utils/dependencies/face_alignment/detection/retinaface/retinaface_detector.py

@@ -26,14 +26,15 @@ class RetinaFaceDetector(FaceDetector):
     '''RetinaFace Detector.
     '''
 
-    def __init__(self, device, path_to_detector=None, verbose=False, threshold=0.5, fp16=True):
+    def __init__(self, device, path_to_detector=None, verbose=False, threshold=0.5, max_size=-1, fp16=True):
         super(RetinaFaceDetector, self).__init__(device, verbose)
 
         # Initialise the face detector
         if path_to_detector is None:
             path_to_detector = self.load_from_url(models_urls['retinaface'], model_dir)
 
         self.threshold = threshold
+        self.max_size = max_size
         self.fp16 = fp16
 
         self.face_detector = load_net(path_to_detector, self.device)
@@ -46,6 +47,7 @@ def detect_from_image(self, tensor_or_path):
         detected_faces = batch_detect(
             self.face_detector,
             [image],
+            max_size=self.max_size,
             threshold=self.threshold,
             return_dict=True,
             fp16=self.fp16,
@@ -60,6 +62,7 @@ def detect_from_batch(self, tensor):
             self.face_detector,
             tensor,
             is_tensor=True,
+            max_size=self.max_size,
             threshold=self.threshold,
             return_dict=True,
             fp16=self.fp16,