I would like to know if there a Pytorch Implementation or library of Connected Components? I am doing cv2.contours on a binary mask on CPU, instead I want to move this to GPU and use connected components.
Any suggestions ?
Thanks
I would like to know if there a Pytorch Implementation or library of Connected Components? I am doing cv2.contours on a binary mask on CPU, instead I want to move this to GPU and use connected components.
Any suggestions ?
Thanks
Kornia
seems to have implemented the connected components labeling algorithm in this PR. CC @edgarriba