U-net model from https://arxiv.org/abs/1505.04597 implemented with additional features:
- residual blocks https://arxiv.org/abs/1512.03385
- dilated convolutions https://arxiv.org/abs/1511.07122
- group normalization https://arxiv.org/abs/1803.08494
Requires PyTorch 0.4