search

koriavinash1 / BraTs2018

Brats segmentation pytorch
14 stars 7 forks source link

How to train the model? #3

Open sunshiding opened 4 years ago

sunshiding commented 4 years ago

There are a lot of files in your folder, how should I use it? Is there an order of use?

koriavinash1 commented 4 years ago

Hi @sunshiding, This repo was used for all our initial experiments if you wish to finetune the model or train from scratch, please check https://github.com/koriavinash1/DeepBrainSeg/blob/master/examples/finetune.py

the example provides information about fine-tuning, it can be modified and used to train any network from scratch also

sunshiding commented 4 years ago

Hi @sunshiding, This repo was used for all our initial experiments if you wish to finetune the model or train from scratch, please check https://github.com/koriavinash1/DeepBrainSeg/blob/master/examples/finetune.py

the example provides information about fine-tuning, it can be modified and used to train any network from scratch also.

Hi, @koriavinash1, Thank you, this project is great. I can start to train the model normally. I am using the Tiramisu_3D project, but I have encountered some problems when predicting. In the Tester.py file:

mask = np.uint8(nib.load(root_path+'get_2D_tiramisu.nii.gz').get_data())

I did not find the code to generate this data. Looking forward to your answers, thank you very much.

koriavinash1 commented 4 years ago

removed that, it was just to limit the iterations for 3D network