Code that implements paper "End-to-End Instance Segmentation with Recurrent Attention".
- Python 2.7
- TensorFlow 0.12 (not compatible with TensorFlow 1.0)
- OpenCV
- NumPy
- SciPy
- PyYaml
- hdf5 and H5Py
- tqdm
- Pillow (required by cityscapes evaluation)
Compile Hungarian matching module
./hungarian_build.shFirst modify setup_cvppp.sh with your dataset folder paths.
./setup_cvppp.shRun experiments:
./run_cvppp.shFirst modify setup_kitti.sh with your dataset folder paths.
./setup_kitti.shRun experiments:
./run_cvppp.shFirst modify setup_cityscapes.sh with your dataset folder paths.
./setup_cityscapes.shRun experiments:
./run_cityscapes.shIf you use our code, please consider cite the following: End-to-End Instance Segmentation with Recurrent Attention. Mengye Ren, Richard S. Zemel. CVPR 2017.
@inproceedings{ren17recattend,
author = {Mengye Ren and Richard S. Zemel},
title = {End-to-End Instance Segmentation with Recurrent Attention},
booktitle = {CVPR},
year = {2017}
}