diff --git a/__init__.py b/__init__.py index 4a7b3fc..4e68d8a 100644 --- a/__init__.py +++ b/__init__.py @@ -3,6 +3,7 @@ from .batch_image_overlay import NODE_CLASS_MAPPINGS as OVERLAY_CLASS_MAPPINGS, from .remove_black_to_transparent import NODE_CLASS_MAPPINGS as REMOVE_BLACK_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as REMOVE_BLACK_DISPLAY_NAMES from .batch_enlarged_overlay import NODE_CLASS_MAPPINGS as ENLARGED_OVERLAY_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as ENLARGED_OVERLAY_DISPLAY_NAMES from .canny_shrink_recenter import NODE_CLASS_MAPPINGS as CANNY_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as CANNY_DISPLAY_NAMES +from .calculate_center_bounding_box import NODE_CLASS_MAPPINGS as CENTER_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as CENTER_DISPLAY_NAMES # Merge all node mappings into one NODE_CLASS_MAPPINGS = { @@ -10,7 +11,8 @@ NODE_CLASS_MAPPINGS = { **OVERLAY_CLASS_MAPPINGS, **REMOVE_BLACK_CLASS_MAPPINGS, **ENLARGED_OVERLAY_CLASS_MAPPINGS, - **CANNY_CLASS_MAPPINGS + **CANNY_CLASS_MAPPINGS, + **CENTER_CLASS_MAPPINGS } NODE_DISPLAY_NAME_MAPPINGS = { @@ -18,8 +20,10 @@ NODE_DISPLAY_NAME_MAPPINGS = { **OVERLAY_DISPLAY_NAMES, **REMOVE_BLACK_DISPLAY_NAMES, **ENLARGED_OVERLAY_DISPLAY_NAMES, - **CANNY_DISPLAY_NAMES + **CANNY_DISPLAY_NAMES, + **CENTER_DISPLAY_NAMES } __all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] +