Image Segmentation With Pytorch