search

nicolov / segmentation_keras

DilatedNet in Keras for image segmentation
MIT License
301 stars 92 forks source link

Context module training implementation plans #29

Open OliverColeman opened 7 years ago

OliverColeman commented 7 years ago

Thanks for creating this implementation. Do you have any plans to implement training of the context module (to allow producing full resolution segmentation maps)?

cclough commented 6 years ago

I'm really interested in training the context module too. Main reason I wish to use this model is for the high res. segmentation prediction.

Is it complex to do - could you guide us on how to do it?

nicolov commented 6 years ago

I don't think it would be complex, but I didn't get around to it back when I implemented this. Do you think this is still a competitive architecture?

cclough commented 6 years ago

Thanks for the quick reply. Yes this fits a niche - it's a better architecture than FCN & SegNet, but doesn't require 8 GPUs to train like the latest 'RefineNet's etc. It's also such nice and simple code in Keras. I can't get, for example DeepLab V2, in Keras....

If you implement it (or guide me on how), I would love to use it.

Christian