Imported from https://github.com/krutikabapat/Sematic_Segmentation_Using_Pytorch