Fix category in ImageContainer
This commit is contained in:
@@ -44,7 +44,7 @@ class ImageContainer:
|
||||
|
||||
RETURN_TYPES = ("IMAGE",)
|
||||
FUNCTION = "image_container"
|
||||
CATEGORY = "mask"
|
||||
CATEGORY = "image"
|
||||
|
||||
def image_container(self, width, height, red, green, blue, alpha):
|
||||
return (create_rgba_image(width, height, (red, green, blue, int(alpha * 255))).image_to_tensor().unsqueeze(0),)
|
||||
@@ -93,7 +93,7 @@ class ImageContainerInheritanceAdd:
|
||||
|
||||
RETURN_TYPES = ("IMAGE",)
|
||||
FUNCTION = "image_container_inheritance_add"
|
||||
CATEGORY = "mask"
|
||||
CATEGORY = "image"
|
||||
|
||||
def image_container_inheritance_add(self, images, add_width, add_height, red, green, blue, alpha, method):
|
||||
width, height = images[0, :, :, 0].shape
|
||||
@@ -156,7 +156,7 @@ class ImageContainerInheritanceScale:
|
||||
|
||||
RETURN_TYPES = ("IMAGE",)
|
||||
FUNCTION = "image_container_inheritance_scale"
|
||||
CATEGORY = "mask"
|
||||
CATEGORY = "image"
|
||||
|
||||
def image_container_inheritance_scale(self, images, scale_width, scale_height, red, green, blue, alpha, method):
|
||||
width, height = images[0, :, :, 0].shape
|
||||
@@ -204,7 +204,7 @@ class ImageContainerInheritanceMax:
|
||||
|
||||
RETURN_TYPES = ("IMAGE",)
|
||||
FUNCTION = "image_container_inheritance_max"
|
||||
CATEGORY = "mask"
|
||||
CATEGORY = "image"
|
||||
|
||||
def image_container_inheritance_max(self, images_a, images_b, red, green, blue, alpha, method):
|
||||
img_a_height, img_a_width = images_a[0, :, :, 0].shape
|
||||
@@ -265,7 +265,7 @@ class ImageContainerInheritanceSum:
|
||||
|
||||
RETURN_TYPES = ("IMAGE",)
|
||||
FUNCTION = "image_container_inheritance_sum"
|
||||
CATEGORY = "mask"
|
||||
CATEGORY = "image"
|
||||
|
||||
def image_container_inheritance_sum(self, images_a, images_b, red, green, blue, alpha, container_size_type, method):
|
||||
img_a_height, img_a_width = images_a[0, :, :, 0].shape
|
||||
|
||||
Reference in New Issue
Block a user