Fixed images/PNG and images/TIFF not existing

Klohger · Aug 1, 2023 · a24151b · a24151b
1 parent f5ee347
commit a24151b
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/MakeBayer.py b/MakeBayer.py
@@ -106,12 +106,15 @@ def SaveAsImage(
     folder = Path("./images")
     folder.mkdir(exist_ok=True)
 
+    pngFolder = folder / "PNG"
+    pngFolder.mkdir(exist_ok=True)
+
     for size in sizes[0:4]:  # 2, 4, 8, 16
         # for tileCount in tileCounts:
         SaveAsImage(
             matrix=matrices[size],
             # tileCount=tileCount,
-            folder=folder / "PNG",
+            folder=pngFolder,
             mode="L",
             sizePadding=3,
         )
@@ -121,17 +124,19 @@ def SaveAsImage(
         SaveAsImage(
             matrix=matrices[size],
             # tileCount=tileCount,
-            folder=folder / "PNG",
+            folder=pngFolder,
             mode="I",
             sizePadding=3,
         )
 
+    tiffFolder = folder / "TIFF"
+    tiffFolder.mkdir(exist_ok=True)
     for size in sizes:
         # for tileCount in tileCounts:
         SaveAsImage(
             matrix=matrices[size],
             # tileCount=tileCount,
-            folder=folder / "TIFF",
+            folder=tiffFolder,
             ext="tiff",
             mode="F",
             sizePadding=4,