How to change the provided segmentation labels of SwinUNETR

[laizhui]

I want to train a SwinUNETR model on my own MRI dataset, using the code and model.pt file given at:
https://github.com/Project-MONAI/research-contributions/tree/main/SwinUNETR/BRATS21
For example, the provided segmentation labels have values of 1 for NCR, 2 for ED, 4 for ET, and 0 for everything else. If I want to use 3 to represent ET,what should I do with the code?
Thank you very much!

[KumoLiu]

Hi @laizhui, thanks for your interest here.
Instead of modifying the weight and model, adjusting your data should be faster and simpler.

MONAI/monai/transforms/utility/array.py

Line 1166 in 95f69de

class MapLabelValue:

Hope it helps, thanks.

[laizhui]

I try to modify my dataset directly, it works.
Thank you very much!