yiff_toolkit / scripts /convert-to-16-bit.py
k4d3's picture
remove resized loras because new ones are coming
7ec09e6
raw
history blame
1.02 kB
import os
import glob
from PIL import Image
def convert_high_bit_depth_images(root_dir, threshold_bit_depth):
print(f"Searching for images in {root_dir} and its subdirectories...")
for filepath in glob.iglob(os.path.join(root_dir, '**/*.(jpg|jpeg|png)'), recursive=True):
print(f"Found image: {filepath}")
with Image.open(filepath) as img:
bit_depth = img.bits
print(f"Image bit depth: {bit_depth} bits")
if bit_depth > threshold_bit_depth:
print(f"Converting image to 16-bit: {filepath}")
img = img.convert('I;16')
img.save(filepath)
print(f"Conversion complete: {filepath}")
else:
print(f"Image bit depth is already 16-bit or lower: {filepath}")
print("Conversion complete!")
def main():
root_dir = 'E:\\training_dir'
threshold_bit_depth = 16
convert_high_bit_depth_images(root_dir, threshold_bit_depth)
if __name__ == "__main__":
main()