File size: 1,408 Bytes
db92a11 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | from .nodes.save_load_pose import TSSavePoseDataAsPickle, TSLoadPoseDataPickle
from .nodes.openpose_smoother import KPSSmoothPoseDataAndRender
from .nodes.load_video_batch import LoadVideoBatchListFromDir
from .nodes.rename_files import RenameFilesInDir
from .nodes.color_match import TSColorMatchSequentialBias
from .nodes.preview_image_metadata import PreviewImageNoMetadata
from .nodes.video_combine_metadata import TSVideoCombineNoMetadata
NODE_CLASS_MAPPINGS = {
"TSSavePoseDataAsPickle": TSSavePoseDataAsPickle,
"TSLoadPoseDataPickle": TSLoadPoseDataPickle,
"TSPoseDataSmoother": KPSSmoothPoseDataAndRender,
"TSLoadVideoBatchListFromDir": LoadVideoBatchListFromDir,
"TSRenameFilesInDir": RenameFilesInDir,
"TSColorMatch": TSColorMatchSequentialBias,
"TSPreviewImageNoMetadata": PreviewImageNoMetadata,
"TSVideoCombineNoMetadata": TSVideoCombineNoMetadata,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"TSSavePoseDataAsPickle": "TS Save Pose Data (PKL)",
"TSLoadPoseDataPickle": "TS Load Pose Data (PKL)",
"TSPoseDataSmoother": "TS Pose Data Smoother",
"TSLoadVideoBatchListFromDir": "TS Load Video Batch List From Dir",
"TSRenameFilesInDir": "TS Rename Files In Dir",
"TSColorMatch": "TS Color Match",
"TSPreviewImageNoMetadata": "TS Preview Image No Metadata",
"TSVideoCombineNoMetadata": "TS Video Combine No Metadata",
}
WEB_DIRECTORY = "web"
|