Imported from https://github.com/massucattoj/unet-model