xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-20 10:14:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add ReverseImageBatch

Browse Source
This commit is contained in:
kijai 2023-11-04 15:28:51 +02:00
parent 1dde3e6210
commit 436b766cba
1 changed files with 45 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
nodes.py
View File

@ -393,6 +393,24 @@ class GetImageRangeFromBatch:
raise ValueError("GetImageRangeFromBatch: End index is out of range") raise ValueError("GetImageRangeFromBatch: End index is out of range")
chosen_images = images[start_index:end_index] chosen_images = images[start_index:end_index]
return (chosen_images, ) return (chosen_images, )
class ReverseImageBatch:
RETURN_TYPES = ("IMAGE",)
FUNCTION = "reverseimagebatch"
CATEGORY = "KJNodes"
@classmethod
def INPUT_TYPES(s):
return {
"required": {
"images": ("IMAGE",),
},
}
def reverseimagebatch(self, images):
reversed_images = torch.flip(images, [0])
return (reversed_images, )
class CreateTextMask: class CreateTextMask:
@ -409,43 +427,46 @@ class CreateTextMask:
"text_x": ("INT", {"default": 0,"min": 0, "max": 4096, "step": 1}), "text_x": ("INT", {"default": 0,"min": 0, "max": 4096, "step": 1}),
"text_y": ("INT", {"default": 0,"min": 0, "max": 4096, "step": 1}), "text_y": ("INT", {"default": 0,"min": 0, "max": 4096, "step": 1}),
"font_size": ("INT", {"default": 32,"min": 8, "max": 4096, "step": 1}), "font_size": ("INT", {"default": 32,"min": 8, "max": 4096, "step": 1}),
"font_color": ("STRING", {"default": "white"}),
"text": ("STRING", {"default": "HELLO!"}), "text": ("STRING", {"default": "HELLO!"}),
"font_path": ("STRING", {"default": "fonts\\TTNorms-Black.otf"}), "font_path": ("STRING", {"default": "fonts\\TTNorms-Black.otf"}),
"width": ("INT", {"default": 256,"min": 16, "max": 4096, "step": 1}), "width": ("INT", {"default": 512,"min": 16, "max": 4096, "step": 1}),
"height": ("INT", {"default": 256,"min": 16, "max": 4096, "step": 1}), "height": ("INT", {"default": 512,"min": 16, "max": 4096, "step": 1}),
"start_rotation": ("INT", {"default": 0,"min": 0, "max": 359, "step": 1}), "start_rotation": ("INT", {"default": 0,"min": 0, "max": 359, "step": 1}),
"end_rotation": ("INT", {"default": 359,"min": -359, "max": 359, "step": 1}), "end_rotation": ("INT", {"default": 0,"min": -359, "max": 359, "step": 1}),
}, },
} }
def createtextmask(self, frames, width, height, invert, text_x, text_y, text, font_size, font_path, start_rotation, end_rotation): def createtextmask(self, frames, width, height, invert, text_x, text_y, text, font_size, font_color, font_path, start_rotation, end_rotation):
# Define the number of images in the batch # Define the number of images in the batch
batch_size = frames batch_size = frames
out = [] out = []
masks = [] masks = []
rotation = start_rotation rotation = start_rotation
if frames > 1: if start_rotation != end_rotation:
rotation_increment = (end_rotation - start_rotation) / (batch_size - 1) rotation_increment = (end_rotation - start_rotation) / (batch_size - 1)
if font_path == "fonts\\TTNorms-Black.otf": #I don't know why relative path won't work otherwise... if font_path == "fonts\\TTNorms-Black.otf": #I don't know why relative path won't work otherwise...
font_path = os.path.join(script_dir, font_path) font_path = os.path.join(script_dir, font_path)
# Generate the text # Generate the text
for i in range(batch_size): for i in range(batch_size):
image = Image.new("RGB", (width, height), "black") image = Image.new("RGB", (width, height), "black")
draw = ImageDraw.Draw(image) draw = ImageDraw.Draw(image)
font = ImageFont.truetype(font_path, font_size) font = ImageFont.truetype(font_path, font_size)
text_width, text_height = draw.textsize(text, font=font) text_width, text_height = draw.textsize(text, font=font)
text_center_x = text_x + text_width / 2 text_center_x = text_x + text_width / 2
text_center_y = text_y + text_height / 2 text_center_y = text_y + text_height / 2
draw.text((text_x, text_y), text, font=font, fill="white") draw.text((text_x, text_y), text, font=font, fill=font_color)
image = image.rotate(rotation, center=(text_center_x, text_center_y)) if start_rotation != end_rotation:
image = np.array(image).astype(np.float32) / 255.0 image = image.rotate(rotation, center=(text_center_x, text_center_y))
image = torch.from_numpy(image)[None,] rotation += rotation_increment
mask = image[:, :, :, 0] image = np.array(image).astype(np.float32) / 255.0
masks.append(mask) image = torch.from_numpy(image)[None,]
out.append(image) mask = image[:, :, :, 0]
rotation += rotation_increment masks.append(mask)
out.append(image)
if invert: if invert:
return (1.0 - torch.cat(out, dim=0),) return (1.0 - torch.cat(out, dim=0), 1.0 - torch.cat(masks, dim=0),)
return (torch.cat(out, dim=0),torch.cat(masks, dim=0),) return (torch.cat(out, dim=0),torch.cat(masks, dim=0),)
class GrowMaskWithBlur: class GrowMaskWithBlur:
@ -1098,7 +1119,8 @@ NODE_CLASS_MAPPINGS = {
"EmptyLatentImagePresets": EmptyLatentImagePresets, "EmptyLatentImagePresets": EmptyLatentImagePresets,
"ColorMatch": ColorMatch, "ColorMatch": ColorMatch,
"GetImageRangeFromBatch": GetImageRangeFromBatch, "GetImageRangeFromBatch": GetImageRangeFromBatch,
"SaveImageWithAlpha": SaveImageWithAlpha "SaveImageWithAlpha": SaveImageWithAlpha,
"ReverseImageBatch": ReverseImageBatch
} }
NODE_DISPLAY_NAME_MAPPINGS = { NODE_DISPLAY_NAME_MAPPINGS = {
"INTConstant": "INT Constant", "INTConstant": "INT Constant",
@ -1120,5 +1142,6 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"EmptyLatentImagePresets": "EmptyLatentImagePresets", "EmptyLatentImagePresets": "EmptyLatentImagePresets",
"ColorMatch": "ColorMatch", "ColorMatch": "ColorMatch",
"GetImageRangeFromBatch": "GetImageRangeFromBatch", "GetImageRangeFromBatch": "GetImageRangeFromBatch",
"SaveImageWithAlpha": "SaveImageWithAlpha" "SaveImageWithAlpha": "SaveImageWithAlpha",
"ReverseImageBatch": "ReverseImageBatch"
} }