Pytorch Image Segmentation Opencv Java