Fix category in ImageContainer

This commit is contained in:
Nourepide
2023-06-04 20:03:16 +03:00
parent 53bbaaeec2
commit 680d46d410
+5 -5
View File
@@ -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