SPNet: Semantic Projection Network for Zero- and Few-Label Semantic Segmentation

Yongqin Xian*, Subhabrata Choudhury*, Yang He, Bernt Schiele, Zeynep Akata. In CVPR 2019
Abstract:

Semantic segmentation is one of the most fundamental problems in computer vision and pixel-level labelling in this context is particularly expensive. Hence, there have been several attempts to reduce the annotation effort such as learning from image level labels and bounding box annotations. In this paper we take this one step further and focus on the challenging task of zero- and few-label learning of semantic segmentation. We define this task as image segmentation by assigning a label to every pixel even though either no labeled sample of that class was present during training, i.e. zero-label semantic segmentation, or only a few labeled samples were present, i.e. few-label semantic segmentation.Our goal is to transfer the knowledge from previously seen classes to novel classes. Our proposed semantic projection network (SPNet) achieves this goal by incorporating class-level semantic information into any net- work designed for semantic segmentation, in an end-to-end manner. We also propose a benchmark for this task on the challenging COCO-Stuff and PASCAL VOC12 datasets. Our model is effective in segmenting novel classes, i.e. alleviating expensive dense annotations, but also in adapting to novel classes without forgetting its prior knowledge, i.e. generalized zero- and few-label semantic segmentation.

Datasets:

Please download COCO-Stuff from here.

For PASCAL please use SBD. For dataset split, keep the files from pascal validation list (val.txt from development kit ) for testing:

mkdir dataset/annotations/val
while read p; do
    mv "SegmentationClassAug/$p.png" dataset/annotation/val
done < val.txt 
Data Splits:

You can find the splits information in data/datasets folder. Please see splits.ipynb for an example on how to use these files.

Initialization Model:

DeepLab initialization should be done using an ImageNet pretrained ResNet. You can use ResNet-101-model.caffemodel from here. Use this to convert it to a pytorch model. A copy of the convert.py file is included with this project.

Hyperparameters:

You can find the hyperparameters used for our paper in the config folder.

Acknowledgement:

Many thanks to Kazuto Nakashima for creating deeplab-pytorch.

Citation:

If you find this useful, please cite our work as follows:

@InProceedings{XianCVPR2019b,
author = {Xian, Yongqin and Choudhury, Subhabrata and He, Yang and Schiele, Bernt and Akata, Zeynep},
title = {Semantic Projection Network for Zero- and Few-Label Semantic Segmentation},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2019}
}