mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-08-22 22:34:25 -07:00
indentation fix
This commit is contained in:
parent
3513a39d93
commit
ef2fb5dab2
1 changed files with 13 additions and 16 deletions
|
@ -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]
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue