Ethanfel
5b10a1a594
Integrate checksum-pinned runtimes, harden interpolation and cleanup paths, and add a SPEED/LDF model-lab workflow.
54 lines
2.3 KiB
Python
54 lines
2.3 KiB
Python
from .nodes import (
|
|
LoadBIMVFIModel, BIMVFIInterpolate, BIMVFISegmentInterpolate, TweenConcatVideos,
|
|
LoadEMAVFIModel, EMAVFIInterpolate, EMAVFISegmentInterpolate,
|
|
LoadSGMVFIModel, SGMVFIInterpolate, SGMVFISegmentInterpolate,
|
|
LoadGIMMVFIModel, GIMMVFIInterpolate, GIMMVFISegmentInterpolate,
|
|
LoadSPEEDVFIModel, SPEEDVFIInterpolate, SPEEDVFISegmentInterpolate,
|
|
LoadLDFVFIModel, LDFVFIInterpolate,
|
|
VFIOptimizer,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"LoadBIMVFIModel": LoadBIMVFIModel,
|
|
"BIMVFIInterpolate": BIMVFIInterpolate,
|
|
"BIMVFISegmentInterpolate": BIMVFISegmentInterpolate,
|
|
"TweenConcatVideos": TweenConcatVideos,
|
|
"LoadEMAVFIModel": LoadEMAVFIModel,
|
|
"EMAVFIInterpolate": EMAVFIInterpolate,
|
|
"EMAVFISegmentInterpolate": EMAVFISegmentInterpolate,
|
|
"LoadSGMVFIModel": LoadSGMVFIModel,
|
|
"SGMVFIInterpolate": SGMVFIInterpolate,
|
|
"SGMVFISegmentInterpolate": SGMVFISegmentInterpolate,
|
|
"LoadGIMMVFIModel": LoadGIMMVFIModel,
|
|
"GIMMVFIInterpolate": GIMMVFIInterpolate,
|
|
"GIMMVFISegmentInterpolate": GIMMVFISegmentInterpolate,
|
|
"LoadSPEEDVFIModel": LoadSPEEDVFIModel,
|
|
"SPEEDVFIInterpolate": SPEEDVFIInterpolate,
|
|
"SPEEDVFISegmentInterpolate": SPEEDVFISegmentInterpolate,
|
|
"LoadLDFVFIModel": LoadLDFVFIModel,
|
|
"LDFVFIInterpolate": LDFVFIInterpolate,
|
|
"VFIOptimizer": VFIOptimizer,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"LoadBIMVFIModel": "Load BIM-VFI Model",
|
|
"BIMVFIInterpolate": "BIM-VFI Interpolate",
|
|
"BIMVFISegmentInterpolate": "BIM-VFI Segment Interpolate",
|
|
"TweenConcatVideos": "Tween Concat Videos",
|
|
"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",
|
|
"LoadGIMMVFIModel": "Load GIMM-VFI Model",
|
|
"GIMMVFIInterpolate": "GIMM-VFI Interpolate",
|
|
"GIMMVFISegmentInterpolate": "GIMM-VFI Segment Interpolate",
|
|
"LoadSPEEDVFIModel": "Load SPEED Model",
|
|
"SPEEDVFIInterpolate": "SPEED Interpolate",
|
|
"SPEEDVFISegmentInterpolate": "SPEED Segment Interpolate",
|
|
"LoadLDFVFIModel": "Load LDF-VFI Model",
|
|
"LDFVFIInterpolate": "LDF-VFI Sequence Interpolate",
|
|
"VFIOptimizer": "VFI Optimizer",
|
|
}
|