zhixuhao / unet

unet for image segmentation
MIT License
4.6k stars 2k forks source link

getting same accuracy on every epoch, please let me know if there is any solution #133

Open avinash2222 opened 5 years ago

avinash2222 commented 5 years ago

i have 4k images, no of epoch=10, adam(.0001) learning rate, binary crass entropy, unet model

tuanphan09 commented 5 years ago

First, I think you should downsampling or crop your image base on your problem.

Second, I used this repo's model to train with my own data and I had some problems on the result. It's unstable, sometime I got good result but most of the time the training stucked at bad local optima.

My solution: