This pytorch code generates segmentation labels of an input image.
Asako Kanezaki. Unsupervised Image Segmentation by Backpropagation. IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), 2018. (pdf)
Note: The extended work has been accepted for publication in IEEE TIP! Please see the code.
python3, pytorch, opencv2, scikit-image
$ python demo.py --input images/101027.jpg