Modified from the up-stream version, this fork modified/enhanced the origin code in the following perspective:
- Added support for batch-inference
- Added support for connected components counting
- Removed 3d labeling (Since I don't need this for my project)
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
Tested on scikit-image example ,follows example.ipynb
> python3 setup.py install
> python3 setup.py test