Referring to the following line: https://github.com/baudcode/tf-semantic-segmentation/blob/master/tf_semantic_segmentation/models/imagenet_unet.py#L31