It would be useful to extend the FCN class to build on ResNet architectures and use pre-trained weights. Useful resources to achieve this * https://www.kaggle.com/meaninglesslives/unet-resnet34-in-keras/notebook * https://github.com/qubvel/segmentation_models