wangzz313 / Connected_components_PyTorch

A PyTorch implementation of Connected Components Labeling

Connected Components Labeling for PyTorch

References

• YACCLAB : Yet Another Connected Components Labeling Benchmark

• Allegretti, Stefano; Bolelli, Federico; Grana, Costantino "Optimized Block-Based Algorithms to Label Connected Components on GPUs." IEEE Transactions on Parallel and Distributed Systems, 2019. BibTex.

---

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