Imported from https://github.com/Linxuxin/Pytorch-UNet-mine