How to Save Results as .nii.gz Instead of PNG?

[Ziyan-Huang]

Great work on this project! I’ve been using it, and it’s working really well. However, I encountered a small issue when running the inference_demo.sh script. By default, it saves the results as PNG images, but I would like to save the segmentation masks in the same size as the original images, in .nii.gz format (NIfTI format).

Could you please guide me on how to modify the script to save the results as .nii.gz files?

I also noticed that your web demo seems to support this feature, but I’m hoping to run this locally. Any help would be appreciated!

Thank you!