Pytorch Image Segmentation Techniques