Awesome
CVPR2020 Memory aggregation networks for efficient interactive video object segmentation
This is the pytorch implementation of the CVPR2020 paper "Memory aggregation networks for efficient interactive video object segmentation".
Preparation
Dependencies
- Python 3.7
- Pytorch 1.0
- Numpy
- tensorboardX
- davisinteractive (Please refer to this link)
Pretrained model
Download deeplabV3+ model pretrained on COCO to this repo.
Dataset
Download DAVIS2017 and scribbles into one folder. Please refer to DAVIS.
If you need the file "DAVIS2017/ImageSets/2017/v_a_l_instances.txt", please refer to the link https://drive.google.com/file/d/1aLPaQ_5lyAi3Lk3d2fOc_xewSrfcrQlc/view?usp=sharing
Train and Test
sh run_local.sh
Evaluation
You can download our model and decompress it for evaluation.
Citation
Please cite this paper in your publications if it helps your research:
@inproceedings{miao2020memory,
title={Memory aggregation networks for efficient interactive video object segmentation},
author={Miao, Jiaxu and Wei, Yunchao and Yang, Yi},
booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
pages={10366--10375},
year={2020}
}