Skip to content

HaodongZhai/Connected_components_PyTorch

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Connected Components Labeling for PyTorch

References


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

img.png

Install

> python3 setup.py install

> python3 setup.py test

About

A PyTorch implementation of Connected Components Labeling

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 88.0%
  • Cuda 8.5%
  • Python 2.4%
  • C++ 1.1%