search

Project-MONAI / model-zoo

MONAI Model Zoo that hosts models in the MONAI Bundle format.
Apache License 2.0
174 stars 65 forks source link

Rotation of output images in prostate_mri_anatomy #481

Closed FaresAlMohamad closed 10 months ago

FaresAlMohamad commented 11 months ago

This problem is about the inference.json config file for the prostate_mri_anatomy model. The transformations applied to the test data during preprocessing are not reversed during postprocessing; thus, the resulting predictions have a different rotation compared to the original images and labels. I would like to add an invertd() function to postprocessing.

Additionally, running the inference on a PC without a GPU leads to an error because the handlers function 'CheckpointLoader' attempts to load the model to the GPU. I want to add a parameter that loads the model to the CPU when the GPU is unavailable.

yiheng-wang-nv commented 11 months ago

Thanks for posting this issue, the enhancements will be helpful. Welcome to contribute the updates, and if you meet any issues when doing the changes, feel free to ask.

yiheng-wang-nv commented 10 months ago

This feature is implemented by @FaresAlMohamad in #482