Imported from https://github.com/shermanhung/U-Net