torchvision.datasets.CocoDetection returns tensors for images a list of tensors for the segmentations in each image. I’m struggling to understand how to work with this for semantic segmentation training.
I think I want to convert this list of segmentations into binary masks, but I’m having trouble figuring out how.
Can somebody help me?