Routine updates.

Author: ATATC

augmentation/transform.py

@@ -86,7 +86,7 @@ def apply(self, img: _npndarray) -> _npndarray:
             smoke_mask[i] = row
         smoke_mask = _repeat(_expand_dims(smoke_mask, -1), 3, 2)
         smoke_mask += smoke_mask[::-1, :, :]
-        smoke_mask += smoke_mask * tex / 255
+        smoke_mask += smoke_mask * tex / 256
         result = _array(img) * (_ones((height, width, 3)) - smoke_mask) + smoke_mask * self._smoke_color
         return _nparray(result)
 

cached_augment.py

@@ -2,7 +2,7 @@
 
 from cv2 import imread, imwrite
 
-from augmentation import TransformBase, Smoke, LowBrightness, Blood
+from augmentation import TransformBase, Smoke
 from utils import get_items
 
 
@@ -18,7 +18,7 @@ def augment_with_structure(src: str, output_dir: str, transform: TransformBase)
 if __name__ == '__main__':
     augment_with_structure("S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/val",
                            "S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/smoke", Smoke())
-    augment_with_structure("S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/val",
-                           "S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/lb", LowBrightness())
-    augment_with_structure("S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/val",
-                           "S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/blood", Blood(20))
+    # augment_with_structure("S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/val",
+    #                        "S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/lb", LowBrightness())
+    # augment_with_structure("S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/val",
+    #                        "S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/blood", Blood(20))

rearrange.py

@@ -17,7 +17,7 @@ def rearrange(src: str, output_dir: str) -> None:
 if __name__ == '__main__':
     rearrange("S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/smoke",
               "S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/smoke_nnunet")
-    rearrange("S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/lb",
-              "S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/lb_nnunet")
-    rearrange("S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/blood",
-              "S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/blood_nnunet")
+    # rearrange("S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/lb",
+    #           "S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/lb_nnunet")
+    # rearrange("S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/blood",
+    #           "S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/blood_nnunet")

validate.py

@@ -1,11 +1,40 @@
-from numpy import ndarray, logical_and
+from typing import Callable
+
+from cv2 import imread, cvtColor, COLOR_RGB2GRAY
+from numpy import ndarray, logical_and, load
+from rich.progress import Progress
+
+from utils import get_items
 
 
 def calculate_dcs(a: ndarray, b: ndarray) -> float:
-    a = a.astype(bool)
-    b = b.astype(bool)
+    a, b = a.astype(bool), b.astype(bool)
     return float(2 * logical_and(a, b).sum() / (a.sum() + b.sum()))
 
 
 def calculate_nsd(a: ndarray, b: ndarray) -> float:
     return abs(a - b).sum() / max(a.sum(), b.sum())
+
+
+def evaluate(src: str, val: str, method: Callable[[ndarray, ndarray], float]) -> float:
+    i = 0
+    r = 0
+    items = get_items(val)
+    with Progress() as progress:
+        task = progress.add_task("[white]Evaluating...", total=int(len(items) / 3))
+        for path in get_items(val):
+            if not path.endswith(".npy"):
+                continue
+            r += method(load(f"{val}/{path}"),
+                        cvtColor(imread(f"{src}/case_{str(i).zfill(4)}.png"), COLOR_RGB2GRAY) / 256)
+            i += 1
+            progress.update(task, advance=1)
+    return r / i
+
+
+if __name__ == '__main__':
+    predicted = "lb"
+    print("DCS:", evaluate(f"S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/{predicted}_predicted",
+                           "S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/val", calculate_dcs))
+    print("NSD:", evaluate(f"S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/{predicted}_predicted",
+                           "S:/SharedDatasets/SegSTRONGC_release/SegSTRONGC_release/val", calculate_nsd))