Pytorch Image Segmentation Opencv Library