doc: correct retarget_eye output size

This commit is contained in:
guojianzhu 2024-07-14 22:47:33 +08:00
parent b370d8059c
commit 1613690d69

View File

@ -199,7 +199,7 @@ class LivePortraitWrapper(object):
"""
kp_source: BxNx3
eye_close_ratio: Bx3
Return: Bx(3*num_kp+2)
Return: Bx(3*num_kp)
"""
feat_eye = concat_feat(kp_source, eye_close_ratio)
@ -212,6 +212,7 @@ class LivePortraitWrapper(object):
"""
kp_source: BxNx3
lip_close_ratio: Bx2
Return: Bx(3*num_kp)
"""
feat_lip = concat_feat(kp_source, lip_close_ratio)