Awesome

Connected Components Labeling for PyTorch

References

• YACCLAB : Yet Another Connected Components Labeling Benchmark

• <p align="justify"> Allegretti, Stefano; Bolelli, Federico; Grana, Costantino "Optimized Block-Based Algorithms to Label Connected Components on GPUs." IEEE Transactions on Parallel and Distributed Systems, 2019. <a title="BibTex" href="http://imagelab.ing.unimore.it/files2/yacclab/YACCLAB_TPDS2019_BibTex.html">BibTex</a>. <a title="Download" href="https://iris.unimore.it/retrieve/handle/11380/1179616/225393/2018_TPDS_Optimized_Block_Based_Algorithms_to_Label_Connected_Components_on_GPUs.pdf"><img src="https://raw.githubusercontent.com/prittt/YACCLAB/master/doc/pdf_logo.png" alt="Download." /></a></p>

Follwing Block-based Union Find Algorithm from YACCLAB.

• Running on GPU.
• PyTorch Interface
• Fix some bit alignment problem
• little refactoring

Example

Tested on scikit-image example ,follows example.ipynb

Install

> python3 setup.py install

> python3 setup.py test