Error happens in segmentation task, but fixed!

MrGiovanni / ModelsGenesis

[MICCAI 2019 Young Scientist Award] [MEDIA 2020 Best Paper Award] Models Genesis

Other

737 stars 140 forks source link

Error happens in segmentation task, but fixed! #10

Closed zem007 closed 4 years ago

zem007 commented 4 years ago

error happens with: _output = activate(finalconvolution, axis=1) cause its output is not a keras layer. suggest to replace with this line: _output = keras.layers.Softmax(axis = 1)(finalconvolution) it works!!!

By the way, good job 纵苇! Your pretrained model improves my project performance.

MrGiovanni commented 4 years ago

Updated, thank you very much for your help.