MONAI icon indicating copy to clipboard operation
MONAI copied to clipboard

Save individual normalization modes (multi-channel) in Auto3dSeg's model config

Open che85 opened this issue 1 year ago • 1 comments

Is your feature request related to a problem? Please describe.

I trained a model with the Auto3dSeg Autorunner and was trying to run inference using the SlicerMONAIAuto3DSeg extension (https://github.com/lassoan/SlicerMONAIAuto3DSeg).

modality: mri
extra_modalities: {image2 : none}    # a second modality is a label which should NOT be normalized

The model expects multiple input images (image1: mri, image2: label --> no normalization).

The following .yaml file is an example of the use of multiple input images for a model: https://github.com/Project-MONAI/tutorials/blob/main/auto3dseg/tasks/hecktor22/input.yaml

Currently, no matter how many input modalities (imageModality and extra_modalities) were provided, only a single value for normalize_mode is saved in the model's `config'.

Describe the solution you'd like

  1. When multiple input images were provided and the same normalization mode was used for all of them, then a single value fornormalize_mode is valid
  2. When multiple input images of different types/modalities were provided, then a list of normalization modes corresponding to the number of input images should be used for normalize_mode.

Additional context

https://github.com/lassoan/SlicerMONAIAuto3DSeg/issues/32 https://github.com/JolleyLab/Internal/issues/194

che85 avatar Mar 26 '24 15:03 che85

cc @myron @dongyang0122 @heyufan1995 for more discussion here.

KumoLiu avatar Apr 17 '24 11:04 KumoLiu