Skip to content

Commit 65d1efa

Browse files
committed
Fix ignored face_alingment parameter
1 parent 4b69cba commit 65d1efa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/simswap.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ def run_detect_align(self, image: np.ndarray, for_id: bool = False) -> Tuple[
134134
kps = kps[None, ...]
135135

136136
align_imgs, transforms = align_face(image, kps, crop_size=self.crop_size,
137-
mode=self.face_alignment_type)
137+
mode='ffhq' if self.face_alignment_type == FaceAlignmentType.FFHQ else 'none')
138138

139139
return align_imgs, transforms, detection.score
140140

0 commit comments

Comments
 (0)