Add SGM-VFI (CVPR 2024) frame interpolation support
SGM-VFI combines local flow estimation with sparse global matching (GMFlow) to handle large motion and occlusion-heavy scenes. Adds 3 new nodes: Load SGM-VFI Model, SGM-VFI Interpolate, SGM-VFI Segment Interpolate. Architecture files vendored from MCG-NJU/SGM-VFI with device-awareness fixes (no hardcoded .cuda()), relative imports, and debug code removed. README updated with model comparison table. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -40,6 +40,7 @@ _auto_install_deps()
|
||||
from .nodes import (
|
||||
LoadBIMVFIModel, BIMVFIInterpolate, BIMVFISegmentInterpolate, BIMVFIConcatVideos,
|
||||
LoadEMAVFIModel, EMAVFIInterpolate, EMAVFISegmentInterpolate,
|
||||
LoadSGMVFIModel, SGMVFIInterpolate, SGMVFISegmentInterpolate,
|
||||
)
|
||||
|
||||
NODE_CLASS_MAPPINGS = {
|
||||
@@ -50,6 +51,9 @@ NODE_CLASS_MAPPINGS = {
|
||||
"LoadEMAVFIModel": LoadEMAVFIModel,
|
||||
"EMAVFIInterpolate": EMAVFIInterpolate,
|
||||
"EMAVFISegmentInterpolate": EMAVFISegmentInterpolate,
|
||||
"LoadSGMVFIModel": LoadSGMVFIModel,
|
||||
"SGMVFIInterpolate": SGMVFIInterpolate,
|
||||
"SGMVFISegmentInterpolate": SGMVFISegmentInterpolate,
|
||||
}
|
||||
|
||||
NODE_DISPLAY_NAME_MAPPINGS = {
|
||||
@@ -60,4 +64,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
|
||||
"LoadEMAVFIModel": "Load EMA-VFI Model",
|
||||
"EMAVFIInterpolate": "EMA-VFI Interpolate",
|
||||
"EMAVFISegmentInterpolate": "EMA-VFI Segment Interpolate",
|
||||
"LoadSGMVFIModel": "Load SGM-VFI Model",
|
||||
"SGMVFIInterpolate": "SGM-VFI Interpolate",
|
||||
"SGMVFISegmentInterpolate": "SGM-VFI Segment Interpolate",
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user