search

ellisdg / 3DUnetCNN

Pytorch 3D U-Net Convolution Neural Network (CNN) designed for medical image segmentation
MIT License
1.9k stars 653 forks source link

The predicted results appear to be incorrect #99

Closed connorchen closed 6 years ago

connorchen commented 6 years ago

hi @ellisdg I use LGG & GBM to generate training and validation data. I use the pre-trained model provided. isensee_2017_model.h5 or tumor_segmentation_model.h5 I run the predict.py But the prediction for tumors is not correct.

Are the pre-trained models you provided correct?

wcfzl commented 6 years ago

I also have the problem,when I use the pre-trained model, isensee_2017_model.h5, it showed:

UserWarning:Error in loading the saved optimizer state. As a result, your model is starting with a freshly initialized optimizer. warnings.warn('Error in loading the saved optimizer

so,I guess that the pre-trained model can't be load,and the prediction comes from a a freshly initialized model,so they are not correct.

can you load the pre-trained model correctly? @alkamid

alkamid commented 6 years ago

@wcfzl Sorry, I didn't try, I trained it on a different dataset.

connorchen commented 6 years ago

Maybe I made a mistake, model trained from train_isensee2017.py give great result