Train on my own dataset

[AmericaBG]

Hi! First of all, thank you very much for sharing your project. It's a great job, congratulations!

I would like to train your network with my own dataset. I've .nii MRI files, Sagital T2 and STIR. Each file (patient) has got several slices, between 12 and 16 (shape=[nslices, 256, 256]).

Could you tell me if would be possible to train it with your network? And, in the event that it is possible, how must I organize the data for training, i.e., all the slices in one array, each file in one array...? For example, if I use 60 patients (.nii files) to train, "mri_data_train" should be of size [60xnslices,256,256]?

Thank you very much in advance!

All the best :)