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