-
Notifications
You must be signed in to change notification settings - Fork 22
/
__init__.py
24 lines (23 loc) · 1.18 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
from .nodes import MarigoldDepthEstimation, MarigoldDepthEstimationVideo, ColorizeDepthmap, SaveImageOpenEXR, RemapDepth
from .nodes_v2 import MarigoldModelLoader, MarigoldDepthEstimation_v2, MarigoldDepthEstimation_v2_video
NODE_CLASS_MAPPINGS = {
"MarigoldModelLoader": MarigoldModelLoader,
"MarigoldDepthEstimation_v2": MarigoldDepthEstimation_v2,
"MarigoldDepthEstimation_v2_video": MarigoldDepthEstimation_v2_video,
"MarigoldDepthEstimation": MarigoldDepthEstimation,
"MarigoldDepthEstimationVideo": MarigoldDepthEstimationVideo,
"ColorizeDepthmap": ColorizeDepthmap,
"SaveImageOpenEXR": SaveImageOpenEXR,
"RemapDepth": RemapDepth
}
NODE_DISPLAY_NAME_MAPPINGS = {
"MarigoldModelLoader": "MarigoldModelLoader",
"MarigoldDepthEstimation_v2": "MarigoldDepthEstimation_v2",
"MarigoldDepthEstimation_v2_video": "MarigoldDepthEstimation_v2_video",
"MarigoldDepthEstimation": "MarigoldDepthEstimation",
"MarigoldDepthEstimationVideo": "MarigoldDepthEstimationVideo",
"ColorizeDepthmap": "Colorize Depthmap",
"SaveImageOpenEXR": "SaveImageOpenEXR",
"RemapDepth": "Remap Depth"
}
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]