# pytorch-unsupervised-segmentation-tip

**Repository Path**: gogotorch/pytorch-unsupervised-segmentation-tip

## Basic Information

- **Project Name**: pytorch-unsupervised-segmentation-tip
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: MIT
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2021-05-03
- **Last Updated**: 2021-05-03

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 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](https://kanezaki.github.io/media/unsupervised_image_segmentation_with_scribbles.png)

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](https://arxiv.org/abs/2007.09990))

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

## What is new?

This is an extension of our [previous work](https://github.com/kanezaki/pytorch-unsupervised-segmentation). 

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