Ethanfel/ComfyUI-Tween
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add EMA-VFI (CVPR 2023) frame interpolation support

Browse Source 
Integrate EMA-VFI alongside existing BIM-VFI with three new ComfyUI nodes:
Load EMA-VFI Model, EMA-VFI Interpolate, and EMA-VFI Segment Interpolate.

Architecture files vendored from MCG-NJU/EMA-VFI with device-awareness
fixes (removed hardcoded .cuda() calls), warp cache management, and
relative imports. InputPadder extended to support EMA-VFI's replicate
center-symmetric padding. Auto-installs timm dependency on first load.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Ethanfel
2026-02-12 22:30:06 +01:00
parent 0133f61d47
commit 1de086569c
11 changed files with 1334 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
utils/padder.py
View File

@@ -4,17 +4,22 @@ import torch.nn.functional as F
class InputPadder:
""" Pads images such that dimensions are divisible by divisor """
def __init__(self, dims, divisor=16):
def __init__(self, dims, divisor=16, mode='constant', center=False):
self.ht, self.wd = dims[-2:]
self.mode = mode
pad_ht = (((self.ht // divisor) + 1) * divisor - self.ht) % divisor
pad_wd = (((self.wd // divisor) + 1) * divisor - self.wd) % divisor
self._pad = [0, pad_wd, 0, pad_ht]
if center:
self._pad = [pad_wd // 2, pad_wd - pad_wd // 2,
pad_ht // 2, pad_ht - pad_ht // 2]
else:
self._pad = [0, pad_wd, 0, pad_ht]
def pad(self, *inputs):
if len(inputs) == 1:
return F.pad(inputs[0], self._pad, mode='constant')
return F.pad(inputs[0], self._pad, mode=self.mode)
else:
return [F.pad(x, self._pad, mode='constant') for x in inputs]
return [F.pad(x, self._pad, mode=self.mode) for x in inputs]
def unpad(self, *inputs):
if len(inputs) == 1: