github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-17 21:02:49 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add inpaint size options to inpaint at a larger size than the actual inpaint image, then scale back down for recombination

Browse Source
This commit is contained in:
Kyle Schouviller
2022-11-05 14:34:52 -07:00
committed by Lincoln Stein
parent 6eeb2107b3
commit 09e41e8f76
2 changed files with 23 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
ldm/invoke/generator/inpaint.py
View File

@@ -150,7 +150,10 @@ class Inpaint(Img2Img):
seam_steps: int = 10,
tile_size: int = 32,
step_callback=None,
inpaint_replace=False, **kwargs):
inpaint_replace=False,
inpaint_width=None,
inpaint_height=None,
**kwargs):
"""
Returns a function returning an image derived from the prompt and
the initial image + mask. Return value depends on the seed at
@@ -168,11 +171,20 @@ class Inpaint(Img2Img):
)
init_filled.paste(init_image, (0,0), init_image.split()[-1])
# Resize if requested for inpainting
if inpaint_width and inpaint_height:
init_filled = init_filled.resize((inpaint_width, inpaint_height))
# Create init tensor
init_image = self._image_to_tensor(init_filled.convert('RGB'))
if isinstance(mask_image, PIL.Image.Image):
self.pil_mask = mask_image
# Resize if requested for inpainting
if inpaint_width and inpaint_height:
mask_image = mask_image.resize((inpaint_width, inpaint_height))
mask_image = mask_image.resize(
(
mask_image.width // downsampling,
@@ -241,6 +253,10 @@ class Inpaint(Img2Img):
result = self.sample_to_image(samples)
# Resize if necessary
if inpaint_width and inpaint_height:
result = result.resize(self.pil_image.size)
# Seam paint if this is our first pass (seam_size set to 0 during seam painting)
if seam_size > 0:
result = self.seam_paint(