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 find_plant.py --input 0184_BP_H_-1_A_211_1595085939000_compressed.JPG
$ python find_plant.py --input 0184_BP_H_-1_A_211_1595085939000_compressed.JPG --scribble
$ python demo_ref.py --input ./BBC/