How to save prediction image

[mineeuk]

I'm using UNETR code a brain image dataset instead of BTCV,
but the prediction image is not saved. Is there any other way?

[KumoLiu]

Hi @ellenyumin, you can use SaveImage to save the predictions.

MONAI/monai/transforms/io/array.py

Line 292 in 97918e4

class SaveImage(Transform):

saver = SaveImage(output_dir="./output", output_ext=".nii.gz", output_postfix="seg")
saver(output)

Thanks!