Hi Everyone, I have a binary image which is the output of a segmentation net. As part of the post processing I have to find all the connected components in this image, which I do with the openCV connectedComponents function. However I was wondering if there is a pytorch equivalent so I can do this operation while the tensor is still on the GPU. Does anyone have or know of something equivalent or similar?
Thanks,
Dave