From 420045cb34380d8a8b4f2ec1a442e87dfd5cf3de Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Wed, 11 Sep 2024 17:42:17 +1000 Subject: [PATCH] feat(nodes): update color map node - Changed name - Better field names --- invokeai/app/invocations/color_map.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/invokeai/app/invocations/color_map.py b/invokeai/app/invocations/color_map.py index 5dbff332d2..e55584caf5 100644 --- a/invokeai/app/invocations/color_map.py +++ b/invokeai/app/invocations/color_map.py @@ -8,17 +8,17 @@ from invokeai.backend.image_util.util import np_to_pil, pil_to_np @invocation( - "color_map_generator", - title="Color Map Generator", + "color_map", + title="Color Map", tags=["controlnet"], category="controlnet", version="1.0.0", ) -class ColorMapGeneratorInvocation(BaseInvocation, WithMetadata, WithBoard): +class ColorMapInvocation(BaseInvocation, WithMetadata, WithBoard): """Generates a color map from the provided image.""" image: ImageField = InputField(description="The image to process") - color_map_tile_size: int = InputField(default=64, ge=1, description=FieldDescriptions.tile_size) + tile_size: int = InputField(default=64, ge=1, description=FieldDescriptions.tile_size) def invoke(self, context: InvocationContext) -> ImageOutput: image = context.images.get_pil(self.image.image_name, "RGB") @@ -26,8 +26,8 @@ class ColorMapGeneratorInvocation(BaseInvocation, WithMetadata, WithBoard): np_image = pil_to_np(image) height, width = np_image.shape[:2] - width_tile_size = min(self.color_map_tile_size, width) - height_tile_size = min(self.color_map_tile_size, height) + width_tile_size = min(self.tile_size, width) + height_tile_size = min(self.tile_size, height) color_map = cv2.resize( np_image,