Fix ignored face_alingment parameter

mike9251 · Jul 15, 2022 · 65d1efa · 65d1efa
1 parent 4b69cba
commit 65d1efa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/simswap.py b/src/simswap.py
@@ -134,7 +134,7 @@ def run_detect_align(self, image: np.ndarray, for_id: bool = False) -> Tuple[
             kps = kps[None, ...]
 
         align_imgs, transforms = align_face(image, kps, crop_size=self.crop_size,
-                                            mode=self.face_alignment_type)
+                                            mode='ffhq' if self.face_alignment_type == FaceAlignmentType.FFHQ else 'none')
 
         return align_imgs, transforms, detection.score