Project-MONAI / research-contributions

Implementations of recent research prototypes/demonstrations using MONAI.
https://monai.io/
Apache License 2.0
1.02k stars 334 forks source link

Offline save predict mask after UNETR #41

Closed ysl2 closed 2 years ago

ysl2 commented 2 years ago

How to offline save predict result after UNETR? Thanks!

And why you caculate dices using reshaped masks? I don't think this is a correct procedure. Instead, this is a trick.

You should save size-restored prediction results offline, and caculate dices with original big masks.

wyli commented 2 years ago

examples are available at https://github.com/Project-MONAI/tutorials/blob/main/3d_segmentation/torch/unet_inference_dict.py