Open Elmiar0642 opened 1 year ago
If you want to use another dataset there are in principle two ways to proceed:
train_unet_neurodata.py
example with it.torch.utils.data.Dataset
and return a dict that contains an 'inp'
key for input data as np.float32
and a 'target'
key for target (label) data as np.int64
arrays if you want to train for classification. Everything else can be freely chosen. Then you can replace the reference to PatchCreator
in the example script by your own Dataset
class.
I wish to use ADNI 3D MRI dataset. I saw training script. Idk how to modify it for custom datasets, especially like ADNI.
I just wish to normally create a model object and then train the model with the 3D images I have inside a directory. I just need to visualise the feature maps at each unit.