para rodar o pytorch pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
This pytorch code generates segmentation labels of an input image.
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.
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)
pytorch, opencv2, tqdm
$ python demo.py --input ./BSD500/101027.jpg
$ python demo.py --input ./PASCAL_VOC_2012/2007_001774.jpg --scribble
$ python demo_ref.py --input ./BBC/