nodes/HelloVision-ComfyUI_HelloMeme
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

lc

Browse Source
This commit is contained in:
songkey-win
2024-11-14 23:43:27 +08:00
parent 8cd8e5677f
commit 045b121242
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hellomeme/utils.py
+3 -2
View File
@@ -121,8 +121,9 @@ face_temp = np.array([[0.50951539, 0.40128625], [0.29259476, 0.40203411], [0.721
def get_drive_pose(toolkits, frames, landmarks, save_size=512, align=True):
if isinstance(landmarks, list): landmarks = np.stack(landmarks, axis=0)
if align:
new_frames, new_landmarks = crop_and_resize(np.stack(frames, axis=0), np.stack(landmarks, axis=0),
new_frames, new_landmarks = crop_and_resize(np.stack(frames, axis=0), landmarks,
save_size=save_size, crop=False)
else:
new_frames, new_landmarks = frames, landmarks
@@ -217,7 +218,7 @@ def det_landmarks(face_aligner, frame_list):
pbar.update()
assert len(save_frame_list) > 0
save_landmark_list = np.stack(save_landmark_list, axis=0).astype(np.float16)
save_landmark_list = np.stack(save_landmark_list, axis=0)
face_aligner.reset_track()
return save_frame_list, save_landmark_list