Connected Components

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