Super-BPD for Fast Image Segmentation (CVPR 2020)

Introduction

We propose direction-based super-BPD, an alternative to superpixel, for fast generic image segmentation, achieving state-of-the-art real-time result.

Citation

Please cite the related works in your publications if it helps your research:


@InProceedings{Wan_2020_CVPR,
author = {Wan, Jianqiang and Liu, Yang and Wei, Donglai and Bai, Xiang and Xu, Yongchao},
title = {Super-BPD: Super Boundary-to-Pixel Direction for Fast Image Segmentation},
booktitle = {The IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2020}
}

Prerequisite

pytorch >= 1.3.0
g++ 7

Dataset

Download the BSDS500 & PascalContext Dataset, and unzip it into the Super-BPD/ folder.

Testing

Compile cuda code for post-process.

cd post_process
python setup.py install

Download the pre-trained PascalContext model and put it in the saved folder.
Test the model and results will be saved in the test_pred_flux/PascalContext folder.
SEISM is used for evaluation of image segmentation.

Training

Download VGG-16 pretrained model.

python train.py --dataset PascalContext