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...?
Thank you very much in advance!

All the best :)