Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

tidy(nodes): rename canvas paste back

Browse Source
This commit is contained in:
psychedelicious 2024-02-24 19:19:29 +11:00
parent cc82ce820a
commit 82249cc634
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
invokeai/app/invocations/image.py
View File

@ -933,14 +933,14 @@ class SaveImageInvocation(BaseInvocation, WithMetadata, WithBoard):
@invocation( @invocation(
"iai_canvas_paste_back", "canvas_paste_back",
title="InvokeAI Canvas Paste Back", title="Canvas Paste Back",
tags=["image", "combine"], tags=["image", "combine"],
category="image", category="image",
version="1.0.0", version="1.0.0",
) )
class IAICanvasPasteBackInvocation(BaseInvocation, WithMetadata, WithBoard): class CanvasPasteBackInvocation(BaseInvocation, WithMetadata, WithBoard):
"""Combines two images by using the mask provided""" """Combines two images by using the mask provided. Intended for use on the Unified Canvas."""
source_image: ImageField = InputField(description="The source image") source_image: ImageField = InputField(description="The source image")
target_image: ImageField = InputField(default=None, description="The target image") target_image: ImageField = InputField(default=None, description="The target image")
@ -949,7 +949,7 @@ class IAICanvasPasteBackInvocation(BaseInvocation, WithMetadata, WithBoard):
) )
mask_blur: int = InputField(default=0, ge=0, description="The amount to blur the mask by") mask_blur: int = InputField(default=0, ge=0, description="The amount to blur the mask by")
def _prepare_mask(self, mask: Image.Image): def _prepare_mask(self, mask: Image.Image) -> Image.Image:
mask_array = numpy.array(mask) mask_array = numpy.array(mask)
kernel = numpy.ones((self.mask_blur, self.mask_blur), numpy.uint8) kernel = numpy.ones((self.mask_blur, self.mask_blur), numpy.uint8)
dilated_mask_array = cv2.erode(mask_array, kernel, iterations=3) dilated_mask_array = cv2.erode(mask_array, kernel, iterations=3)