github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix: Handle cases where tile size > image size

Browse Source
This commit is contained in:
blessedcoolant
2023-09-23 02:18:59 +05:30
committed by psychedelicious
parent f718de98f8
commit 021f77e35e
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/app/invocations/controlnet_image_processors.py
View File

@@ -578,11 +578,14 @@ class ColorMapImageProcessorInvocation(ImageProcessorInvocation):
image = np.array(image, dtype=np.uint8)
height, width = image.shape[:2]
width_tile_size = min(self.color_map_tile_size, width)
height_tile_size = min(self.color_map_tile_size, height)
color_map = cv2.resize(
image,
(width // self.color_map_tile_size, height // self.color_map_tile_size),
(width // width_tile_size, height // height_tile_size),
interpolation=cv2.INTER_CUBIC,
)
color_map = cv2.resize(color_map, (width, height), interpolation=cv2.INTER_NEAREST)
color_map = Image.fromarray(color_map)
color_map = Image.fromarray(color_map)
return color_map