github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 14:24:40 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

indentation fix

Browse source
This commit is contained in:
Jeremy Hummel 2019-08-15 14:51:52 -07:00
commit ef2fb5dab2
1 changed files with 13 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

29
facelib/LandmarksProcessor.py
View file

@ -124,13 +124,10 @@ def get_transform_mat (image_landmarks, output_size, face_type, scale=1.0):
""" """
if face_type == FaceType.AVATAR: if face_type == FaceType.AVATAR:
centroid = np.mean (image_landmarks, axis=0) centroid = np.mean (image_landmarks, axis=0)
mat = umeyama(image_landmarks[17:], landmarks_2D, True)[0:2] mat = umeyama(image_landmarks[17:], landmarks_2D, True)[0:2]
a, c = mat[0,0], mat[1,0] a, c = mat[0,0], mat[1,0]
scale = math.sqrt((a * a) + (c * c)) scale = math.sqrt((a * a) + (c * c))
padding = (output_size / 64) * 32 padding = (output_size / 64) * 32
mat = np.eye ( 2,3 ) mat = np.eye ( 2,3 )
mat[0,2] = -centroid[0] mat[0,2] = -centroid[0]
mat[1,2] = -centroid[1] mat[1,2] = -centroid[1]
@ -143,20 +140,20 @@ def get_transform_mat (image_landmarks, output_size, face_type, scale=1.0):
face_type = FaceType.FULL face_type = FaceType.FULL
remove_align = True remove_align = True
if face_type == FaceType.HALF: if face_type == FaceType.HALF:
padding = 0 padding = 0
elif face_type == FaceType.FULL: elif face_type == FaceType.FULL:
padding = (output_size / 64) * 12 padding = (output_size / 64) * 12
elif face_type == FaceType.HEAD: elif face_type == FaceType.HEAD:
padding = (output_size / 64) * 24 padding = (output_size / 64) * 24
else: else:
raise ValueError ('wrong face_type: ', face_type) raise ValueError ('wrong face_type: ', face_type)
mat = umeyama(image_landmarks[17:], landmarks_2D, True)[0:2] mat = umeyama(image_landmarks[17:], landmarks_2D, True)[0:2]
mat = mat * (output_size - 2 * padding) mat = mat * (output_size - 2 * padding)
mat[:,2] += padding mat[:,2] += padding
mat *= (1 / scale) mat *= (1 / scale)
mat[:,2] += -output_size*( ( (1 / scale) - 1.0 ) / 2 ) mat[:,2] += -output_size*( ( (1 / scale) - 1.0 ) / 2 )
if remove_align: if remove_align:
bbox = transform_points ( [ (0,0), (0,output_size-1), (output_size-1, output_size-1), (output_size-1,0) ], mat, True) bbox = transform_points ( [ (0,0), (0,output_size-1), (output_size-1, output_size-1), (output_size-1,0) ], mat, True)