david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed face jitter

Browse Source
This commit is contained in:
Colombo
2020-03-18 13:40:06 +04:00
parent 8a622428b1
commit 9b070e10a6
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facelib/LandmarksProcessor.py
+10 -10
View File
@@ -281,15 +281,15 @@ def get_transform_mat (image_landmarks, output_size, face_type, scale=1.0):
# calc 3 points in global space to estimate 2d affine transform # calc 3 points in global space to estimate 2d affine transform
if not remove_align: if not remove_align:
l_t = np.array( [ np.round( g_c - tb_diag_vec*mod ), l_t = np.array( [ g_c - tb_diag_vec*mod,
np.round( g_c + bt_diag_vec*mod ), g_c + bt_diag_vec*mod,
np.round( g_c + tb_diag_vec*mod ) ] ) g_c + tb_diag_vec*mod ] )
else: else:
# remove_align - face will be centered in the frame but not aligned # remove_align - face will be centered in the frame but not aligned
l_t = np.array( [ np.round( g_c - tb_diag_vec*mod ), l_t = np.array( [ g_c - tb_diag_vec*mod,
np.round( g_c + bt_diag_vec*mod ), g_c + bt_diag_vec*mod,
np.round( g_c + tb_diag_vec*mod ), g_c + tb_diag_vec*mod,
np.round( g_c - bt_diag_vec*mod ), g_c - bt_diag_vec*mod,
] ) ] )
# get area of face square in global space # get area of face square in global space
@@ -299,9 +299,9 @@ def get_transform_mat (image_landmarks, output_size, face_type, scale=1.0):
side = np.float32(math.sqrt(area) / 2) side = np.float32(math.sqrt(area) / 2)
# calc 3 points with unrotated square # calc 3 points with unrotated square
l_t = np.array( [ np.round( g_c + [-side,-side] ), l_t = np.array( [ g_c + [-side,-side],
np.round( g_c + [ side,-side] ), g_c + [ side,-side],
np.round( g_c + [ side, side] ) ] ) g_c + [ side, side] ] )
# calc affine transform from 3 global space points to 3 local space points size of 'output_size' # calc affine transform from 3 global space points to 3 local space points size of 'output_size'
pts2 = np.float32(( (0,0),(output_size,0),(output_size,output_size) )) pts2 = np.float32(( (0,0),(output_size,0),(output_size,output_size) ))