No description, website, or topics provided.
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
Alexnet init release Dec 5, 2017
VGG minor Dec 5, 2017
caffe init release Dec 5, 2017
data init release Dec 5, 2017
resources remove unrelated code Dec 5, 2017
.gitattributes Create .gitattributes Apr 24, 2018
.gitignore remove pyc Dec 5, 2017
LICENSE Create LICENSE Dec 20, 2017
README.md acknowledgement Dec 13, 2017

README.md

Implementation of Mix-and-Match Tuning for Self-Supervised Semantic Segmentation.

Paper

Xiaohang Zhan, Ziwei Liu, Ping Luo, Xiaoou Tang, Chen Change Loy, "Mix-and-Match Tuning for Self-Supervised Semantic Segmentation", AAAI 2018

Project Page: link

Dependency

Library (Note that the versions are not strictly restricted): OpenMPI=1.8.5, CUDA=8.0, CUDNN=5.1.10

Python: cv2

Before Start

  1. Download pre-trained models in link to pretrain. (You can also find trained graph_iter_xxx.caffemodel which can directly be fine-tuned for segmentation.)

  2. Download PASCAL VOC 2012 augmented dataset and CityScapes dataset to a proper position.

    For PASCAL VOC 2012, create standard training list as shown in data/pascal/train.txt and validation list as shown in data/pascal/val.txt

    For CityScapes, create standard training list as shown in data/cityscapes/train.txt and validation list as shown in data/cityscapes/val.txt

  3. Build caffe with cmake

    cd caffe
    sh build.sh
    

training

For example, train alexnet with colorization as pretrained model.

cd Alexnet/colorize

Then edit train_graph.prototxt and finetune_seg.prototxt to specify "source" and "root_dir" in the data layer.

sh run_graph.sh # or use trained models in the Google Drive folder.
sh run_seg.sh

testing

Edit test.sh to specify data root, testing list and ground truth root.

sh test.sh

Testing results are saved in snapshot/seg_iter_xxx/ by default.

Citing Mix-and-Match

@inproceedings{zhan2018m&m,
 author = {Xiaohang Zhan, Ziwei Liu, Ping Luo, Xiaoou Tang, and Chen Change Loy},
 title = {Mix-and-Match Tuning for Self-Supervised Semantic Segmentation},
 booktitle = {AAAI Conference on Artificial Intelligence (AAAI)},
 month = {February},
 year = {2018} 
}

Acknowledgement

The CAFFE is forked from https://github.com/yjxiong/caffe