This project is derived from the original work in the Nvidia short course on image segmentation