Shapefit: Turn irregular shape masks into clean polygons

I've encountered this problem many times and tried to solve it with different computer vision and algebra methods. Since it's easy to generate clean polygons and augment them a NN can be trained to undo it.