diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-12-25 12:44:15 +0000 |
|---|---|---|
| committer | Aarni Koskela <akx@iki.fi> | 2023-12-30 14:24:01 +0000 |
| commit | 7aa27b000a3087dcb5cc7254600064bf70cacd3e (patch) | |
| tree | 5b2e6bfcab4e1e7829d7057dc0f53f20b5dcb06a /modules/images.py | |
| parent | 31992eff9b9714c158b12cec16dfe66c76270dfa (diff) | |
| download | stable-diffusion-webui-gfx803-7aa27b000a3087dcb5cc7254600064bf70cacd3e.tar.gz stable-diffusion-webui-gfx803-7aa27b000a3087dcb5cc7254600064bf70cacd3e.tar.bz2 stable-diffusion-webui-gfx803-7aa27b000a3087dcb5cc7254600064bf70cacd3e.zip | |
Add types to split_grid
Diffstat (limited to 'modules/images.py')
| -rw-r--r-- | modules/images.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/modules/images.py b/modules/images.py index 16f9ae7c..d30e8865 100644 --- a/modules/images.py +++ b/modules/images.py @@ -64,9 +64,8 @@ def image_grid(imgs, batch_size=1, rows=None): Grid = namedtuple("Grid", ["tiles", "tile_w", "tile_h", "image_w", "image_h", "overlap"])
-def split_grid(image, tile_w=512, tile_h=512, overlap=64):
- w = image.width
- h = image.height
+def split_grid(image: Image.Image, tile_w: int = 512, tile_h: int = 512, overlap: int = 64) -> Grid:
+ w, h = image.size
non_overlap_width = tile_w - overlap
non_overlap_height = tile_h - overlap
|