Resize images based on Pixel amount to optimize max size to vram usage and to keep the original pics uncropped

[spechok]

first ever something something

lets get started:

import os
from PIL import Image
import math

def resize_image(img_path, max_pixels):
# Open the image

img = Image.open(img_path)

# Get the image size
width, height = img.size

# Check if total pixels are over max_pixels

if width * height > max_pixels:
    # Calculate the new size to keep aspect ratio
    aspect_ratio = width / height
    new_height = int(math.sqrt(max_pixels / aspect_ratio))
    new_width = int(new_height * aspect_ratio)
    
    # Resize the image
    img = img.resize((new_width, new_height))
    
    # Save the resized image
    img.save("resized_" + img_path)

Maximum number of pixels

max_pixels = 600000

#you're gonna need to know what is your max pixel amount is, for me it is 6gbvram - so 768*768 =~590,000 from tests i could use up to #~600,000 pixels

Folder name

folder_name = r"E:\stable_diffusion\1"

Loop through all files in the folder

for filename in os.listdir(folder_name):
# Get the file path
file_path = os.path.join(folder_name, filename)

# Check if the file is an image

if file_path.endswith(".jpg") or file_path.endswith(".jpeg") or file_path.endswith(".png"):
    # Call the resize_image function
    resize_image(file_path, max_pixels)

and this is it...
this code basically will take your images and resize them with the same Vertical to Horizontal ratio but to the right resolution
dunno how to implement it into the gradio sadly

anyway, hope ya like it

[spechok]

another option that comes in with darkening transparent PNG pixels to full black(0,0,0,255)
can be also white(255,255,255,255)

import os
from PIL import Image
import math

def resize_image(img_path, max_pixels):
# Open the image
img = Image.open(img_path)

# Check if the image is a PNG and has transparency
if img.mode == 'RGBA':
    # Create a solid black image to paste over the transparent areas
    black_background = Image.new('RGBA', img.size, (0, 0, 0, 255))
    # Paste the original image over the black image
    black_background.paste(img, mask=img)
    img = black_background

# Get the image size
width, height = img.size

# Check if total pixels are over max_pixels
if width * height > max_pixels:
    # Calculate the new size to keep aspect ratio
    aspect_ratio = width / height
    new_height = int(math.sqrt(max_pixels / aspect_ratio))
    new_width = int(new_height * aspect_ratio)

    # Resize the image
    img = img.resize((new_width, new_height))

    # Save the resized image
    img.save( img_path)

Maximum number of pixels

max_pixels = 600000

Folder name

folder_name = r"E:\stable_diffusion\1"

Loop through all files in the folder

for filename in os.listdir(folder_name):
# Get the file path
file_path = os.path.join(folder_name, filename)

# Check if the file is an image
if file_path.endswith(".jpg") or file_path.endswith(".jpeg") or file_path.endswith(".png"):
    # Call the resize_image function
    resize_image(file_path, max_pixels)