diff --git a/nnunetv2/preprocessing/normalization/map_channel_name_to_normalization.py b/nnunetv2/preprocessing/normalization/map_channel_name_to_normalization.py index 18f027bed..7d92b56a7 100644 --- a/nnunetv2/preprocessing/normalization/map_channel_name_to_normalization.py +++ b/nnunetv2/preprocessing/normalization/map_channel_name_to_normalization.py @@ -4,8 +4,8 @@ ZScoreNormalization, RescaleTo01Normalization, RGBTo01Normalization, ImageNormalization channel_name_to_normalization_mapping = { - 'CT': CTNormalization, - 'noNorm': NoNormalization, + 'ct': CTNormalization, + 'nonorm': NoNormalization, 'zscore': ZScoreNormalization, 'rescale_to_0_1': RescaleTo01Normalization, 'rgb_to_0_1': RGBTo01Normalization @@ -17,7 +17,7 @@ def get_normalization_scheme(channel_name: str) -> Type[ImageNormalization]: If we find the channel_name in channel_name_to_normalization_mapping return the corresponding normalization. If it is not found, use the default (ZScoreNormalization) """ - norm_scheme = channel_name_to_normalization_mapping.get(channel_name) + norm_scheme = channel_name_to_normalization_mapping.get(channel_name.casefold()) if norm_scheme is None: norm_scheme = ZScoreNormalization # print('Using %s for image normalization' % norm_scheme.__name__)