https://medium.com/@monocosmo77/working-with-unets-in-different-scenarios-part4-machine-learning-6f2f68989f3d