Pytorch Image Segmentation Opencv C