nodes/FuouM-ComfyUI-FirstOrderMM
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix find_best_frame

Browse Source 
used to take the number of color channels (3) instead of number of frames
This commit is contained in:
Fuou Marinas
2024-06-30 22:26:08 +07:00
committed by GitHub
parent 2d906f4af8
commit dfcb5203dc
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils.py
+2 -2
View File
@@ -120,7 +120,7 @@ def tensor_to_numpy(tensor):
return tensor.squeeze(0).permute(1, 2, 0).cpu().numpy()
def find_best_frame(source_image, driving_video: list, cpu=False) -> int:
def find_best_frame(source_image, driving_video, cpu=False) -> int:
if not face_alignment_available:
print("Face Alignment module were not found. Best frame set to 0.")
return 0
@@ -139,7 +139,7 @@ def find_best_frame(source_image, driving_video: list, cpu=False) -> int:
norm = float("inf")
frame_num = 0
num_frames = driving_video.shape[1]
num_frames = driving_video.shape[2]
for i in tqdm(range(num_frames)):
# Extract single frame from driving video