Add All Folders
This commit is contained in:
@@ -0,0 +1,24 @@
|
||||
# app/gui/napari_main.py
|
||||
import napari
|
||||
from pathlib import Path
|
||||
import SimpleITK as sitk
|
||||
import numpy as np
|
||||
|
||||
def sitk_to_numpy(img):
|
||||
return sitk.GetArrayFromImage(img) # z, y, x
|
||||
|
||||
def run_napari_viewer():
|
||||
viewer = napari.Viewer(title="Radiuma Mini - Viewer")
|
||||
# Load outputs if they exist
|
||||
filtered = Path("app/storage/artifacts/filtered.nii.gz")
|
||||
mask = Path("data/masks/CT_AVM_mask.nii.gz")
|
||||
if filtered.exists():
|
||||
arr = sitk_to_numpy(sitk.ReadImage(str(filtered)))
|
||||
viewer.add_image(arr, name="filtered", blending="translucent")
|
||||
if mask.exists():
|
||||
marr = sitk_to_numpy(sitk.ReadImage(str(mask)))
|
||||
viewer.add_labels(marr, name="mask")
|
||||
napari.run()
|
||||
|
||||
if __name__ == "__main__":
|
||||
run_napari_viewer()
|
||||
Reference in New Issue
Block a user