awoo

Signed-off-by: Balazs Horvath <[email protected]>

scripts/convert-to-16-bit.py

@@ -0,0 +1,48 @@
+import glob
+import os
+from PIL import Image
+import multiprocessing as mp
+
+def convert_image(image_path):
+    try:
+        # Open the image
+        image = Image.open(image_path)
+
+        # Get the image mode
+        mode = image.mode
+
+        # Check if the image depth is higher than 16-bit
+        if mode == 'I;16' or mode == 'I;16L' or mode == 'I;16B':
+            # Image is already 16-bit or lower, skip conversion
+            return
+
+        # Convert the image to 16-bit depth
+        image = image.convert('I')
+
+        # Save the converted image
+        image.save(image_path)
+        print(f"Converted {image_path} to 16-bit.")
+
+    except Exception as e:
+        print(f"Error converting {image_path}: {e}")
+
+def process_directory(root_dir):
+    # Create a list of image file paths
+    image_files = []
+    for ext in ['.png', '.jpg', '.jpeg']:
+        image_files.extend(glob.glob(os.path.join(root_dir, '**', f'*{ext}'), recursive=True))
+
+    # Create a pool of worker processes
+    pool = mp.Pool(processes=mp.cpu_count())
+
+    # Convert each image in parallel
+    pool.map(convert_image, image_files)
+
+    # Close the pool
+    pool.close()
+    pool.join()
+
+if __name__ == '__main__':
+    #root_dir = r'E:\training_dir_staging'
+    root_dir = r'E:\training_dir'
+    process_directory(root_dir)

tests/realistic-v5-candidate1.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6096f1ec8340c6e5b28421e80e265374b9db1870c26cc6c202d45fe745dcf57f
+size 240606160