From 2bf50904b08f0fd1899ac0d7a7e19348e2981574 Mon Sep 17 00:00:00 2001 From: LonicaMewinsky <93007558+LonicaMewinsky@users.noreply.github.com> Date: Tue, 14 Nov 2023 14:21:17 -0600 Subject: [PATCH] keyframer now also appends last frame --- nodes/MakeFrame.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nodes/MakeFrame.py b/nodes/MakeFrame.py index 8902122..8951e85 100644 --- a/nodes/MakeFrame.py +++ b/nodes/MakeFrame.py @@ -78,12 +78,13 @@ class GetKeyFrames: CATEGORY = "Frames" def getkeyframes(self, frames, num_keyframes): - N = np.clip(num_keyframes, 2, len(frames)-1) # Save a spot for first frame + N = np.clip(num_keyframes, 2, len(frames)-2) frames = frames differences = [torch.norm(frames[i+1] - frames[i], p=2) for i in range(len(frames)-1)] _, top_indices = torch.topk(torch.tensor(differences), k=N, largest=True) keyframe_indices = sorted([index.item() + 1 for index in top_indices]) keyframe_indices.insert(0, 0) + keyframe_indices.append(len(frames)-1) cat_keyframe_tensors = [frames[i].to(device).unsqueeze(0) for i in keyframe_indices] cat_keyframe_tensors = torch.cat(cat_keyframe_tensors, dim = 0).to(device)