Skip to content

chayryali/pytorch-unsupervised-segmentation-tip

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Unsupervised Learning of Image Segmentation Based on Differentiable Feature Clustering

This pytorch code generates segmentation labels of an input image.

Unsupervised Image Segmentation with Scribbles

Wonjik Kim*, Asako Kanezaki*, and Masayuki Tanaka. Unsupervised Learning of Image Segmentation Based on Differentiable Feature Clustering. IEEE Transactions on Image Processing, accepted, 2020. (arXiv)

*W. Kim and A. Kanezaki contributed equally to this work.

What is new?

This is an extension of our previous work.

  • Better performance with spatial continuity loss
  • Option of using scribbles as user input
  • Option of using reference image(s)

Requirements

pytorch, opencv2, tqdm

Getting started

Vanilla

$ python demo.py --input ./BSD500/101027.jpg

Vanilla + scribbles

$ python demo.py --input ./PASCAL_VOC_2012/2007_001774.jpg --scribble

Vanilla + reference image(s)

$ python demo_ref.py --input ./BBC/

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%