Routine updates.

Author: ATATC

cached_augment.py

@@ -1,26 +1,16 @@
-from os import listdir, makedirs
-from os.path import isdir
+from os import makedirs
 
 from cv2 import imread, imwrite
 
 from augmentation import TransformBase, Smoke, LowBrightness, Blood
-
-
-def get_items(src: str, branch: str = "") -> str | list[str]:
-    if not isdir(target := f"{src}/{branch}"):
-        return f"{branch}" if target.endswith(".png") else []
-    r = []
-    for f in listdir(target):
-        if isinstance(item := get_items(src, f"{branch}/{f}"), str):
-            r.append(item)
-        else:
-            r += item
-    return r
+from utils import get_items
 
 
 def augment_with_structure(src: str, output_dir: str, transform: TransformBase) -> None:
     makedirs(output_dir, exist_ok=True)
     for path in get_items(src):
+        if not path.endswith(".png"):
+            continue
         makedirs(f"{output_dir}/{path[:path.rfind('/')]}", exist_ok=True)
         imwrite(f"{output_dir}/{path}", transform(imread(f"{src}/{path}")))
 

npy2png.py

@@ -0,0 +1,16 @@
+from cv2 import imwrite
+from numpy import load, uint8
+
+from utils import get_items
+
+
+def npy2png(src: str, output_dir: str) -> None:
+    for path in get_items(src):
+        if not path.endswith('.npy'):
+            continue
+        imwrite(f"{output_dir}/{path.replace('.npy', '.png')}", load(f"{src}/{path}").astype(uint8))
+
+
+if __name__ == '__main__':
+    npy2png("S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/smoke/1/0",
+            "S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/smoke/1/png")

rearrange.py

@@ -0,0 +1,17 @@
+from shutil import copyfile
+
+from utils import get_items
+
+
+def rearrange(src: str, output_dir: str) -> None:
+    i = 0
+    for path in get_items(src):
+        if not path.endswith(".png"):
+            continue
+        copyfile(f"{src}/{path}", f"{output_dir}/case_{str(i).zfill(4)}_0000.png")
+        i += 1
+
+
+if __name__ == '__main__':
+    rearrange("S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/smoke",
+              "S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/smoke_nnunet")

utils.py

@@ -0,0 +1,14 @@
+from os import listdir
+from os.path import isdir
+
+
+def get_items(src: str, branch: str = "") -> str | list[str]:
+    if not isdir(target := f"{src}/{branch}"):
+        return f"{branch}"
+    r = []
+    for f in listdir(target):
+        if isinstance(item := get_items(src, f"{branch}/{f}"), str):
+            r.append(item)
+        else:
+            r += item
+    return r