My hdf5 files still segments membrane dataset

zhixuhao / unet

unet for image segmentation

MIT License

4.6k stars 2k forks source link

My hdf5 files still segments membrane dataset #171

Closed metobom closed 4 years ago

metobom commented 4 years ago

Hello I trained my own dataset but when I predict images I get this output: 3_predict (2000 steps, 5 epoch) Unexpectedly, when I predict membrane dataset I get pretty good results. Any ideas?

metobom commented 4 years ago

I solved my problem by preprocessing my images and increasing my image number from 29 to 34 (68 with augmentation).