Connected Components Labeling for PyTorch
October 12, 2020 ยท View on GitHub
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