Pytorch Image Segmentation Opencv