also U-net is great at this stuff, and a bit easier to implement for my taste. there are many ready implementations over the internet