DiffusionAi/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2025-04-04 22:43:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(nodes): update content shuffle node

Browse Source 
- Better field names
This commit is contained in:
psychedelicious 2024-09-11 17:42:55 +10:00 committed by Kent Keirsey
parent 420045cb34
commit 67a234c1bb
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/app/invocations/content_shuffle.py
View File

@ -16,10 +16,10 @@ class ContentShuffleInvocation(BaseInvocation, WithMetadata, WithBoard):
"""Shuffles the image, similar to a 'liquify' filter."""
image: ImageField = InputField(description="The image to process")
f: int = InputField(default=256, ge=0, description="Content shuffle `f` parameter")
scale_factor: int = InputField(default=256, ge=0, description="The scale factor used for the shuffle")
def invoke(self, context: InvocationContext) -> ImageOutput:
image = context.images.get_pil(self.image.image_name, "RGB")
output_image = content_shuffle(input_image=image, f=self.f)
output_image = content_shuffle(input_image=image, scale_factor=self.scale_factor)
image_dto = context.images.save(image=output_image)
return ImageOutput.build(image_dto)

10
invokeai/backend/image_util/content_shuffle.py
View File

@ -18,18 +18,18 @@ def make_noise_disk(H, W, C, F):
return noise
def content_shuffle(input_image: Image.Image, f: int | None = None) -> Image.Image:
def content_shuffle(input_image: Image.Image, scale_factor: int | None = None) -> Image.Image:
"""Shuffles the content of an image using a disk noise pattern, similar to a 'liquify' effect."""
np_img = pil_to_np(input_image)
height, width, _channels = np_img.shape
if f is None:
f = 256
if scale_factor is None:
scale_factor = 256
x = make_noise_disk(height, width, 1, f) * float(width - 1)
y = make_noise_disk(height, width, 1, f) * float(height - 1)
x = make_noise_disk(height, width, 1, scale_factor) * float(width - 1)
y = make_noise_disk(height, width, 1, scale_factor) * float(height - 1)
flow = np.concatenate([x, y], axis=2).astype(np.float32)