Add Thin-Plate Spline Motion Model for Image Animation (#6)

2024-08-04 15:53:09 +07:00
parent cd43aad974
commit f1f1e37952
21 changed files with 2263 additions and 14 deletions
@@ -5,6 +5,7 @@ from .run import (
    FOMM_Seg10Chooser,
    FOMM_Seg15Chooser,
    Articulate_Runner,
+    Spline_Runner
 )

 NODE_CLASS_MAPPINGS = {
@@ -14,6 +15,7 @@ NODE_CLASS_MAPPINGS = {
    "FOMM_Seg10Chooser": FOMM_Seg10Chooser,
    "FOMM_Seg15Chooser": FOMM_Seg15Chooser,
    "Articulate_Runner": Articulate_Runner,
+    "Spline_Runner": Spline_Runner,
 }

 NODE_DISPLAY_NAME_MAPPINGS = {
@@ -23,6 +25,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
    "FOMM_Seg10Chooser": "FOMM Seg10 Chooser",
    "FOMM_Seg15Chooser": "FOMM Seg15 Chooser",
    "Articulate_Runner": "Articulate Runner",
+    "Spline_Runner": "Spline Runner",
 }