IVD-Net
IVD-Net copied to clipboard
Train with two types of multimodal MRI
Hi Jose! First of all, thank you very much for sharing your code, it's a great job.
I would like to try it in my own data that is composed with STIR and T2 MRI. Is it possible? Should I change some parts of the code?
Thank you very much in advance.
Hi, to adapt to 2 modalities you need to modify the code of the IVD_Net_asym, which is quite easy. Basically you have to change the torch.cat operations from 4 to 2 modalities and adapt the blocks shape accordingly.
Jose