Skip to content

zsef123/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