Hello there,
I build myself a Unet accordingly to the original Unet paper with input 564x564 RGB and smaller output segmentation map: GitHub - CaipiDE/unet_mirror 3
I also implemented the mirroring strategy from the paper. So given the fact that I wanted to have a segmentation mask as big as my input, my mirror_extrapolate algorithm mirrors each side with 98px. As a result I now insert a 756x756 input (image with mirroring sides) which results in a 564x564 segmentation map, perfect 
I testet it on the inria satellite imagary labeling dataset for building detection and was surprised how good it looked after just the first epoch!

vienna5_png564×564 672 KB


epoch_0_mask564×564 9.15 KB

But with increasing epochs, the map just literally disappears?  The segmented objects have no sharp edges and are just some wobbely floating random points (not complete random but you get the point)…

Capture1419×600 346 KB

Did anyone else maybe faced of this behaviour when they did unet or maybe its typicall for a mistake I made?
PS: here is again the Unet structure and my mirroring stuff how that works:

u-net-architecture1555×1036 74.5 KB