xiaomengyc / SG-One

SG-One: Similarity Guidance Network for One-Shot Semantic Segmentation

SG-One: Similarity Guidance Network for One-Shot Semantic Segmentation

Overview of SG-One

Train

You can use the scripts in scripts to train your model on different splits of the VOC2012 dataset. For example,

cd scripts
base train_group0.sh

Test

You can use val_frame_all_mlclasses.py to test your trained models.

Citation

If you find this code helpful, please consider to cite this paper:

@article{zhang2020sg,
  title={Sg-one: Similarity guidance network for one-shot semantic segmentation},
  author={Zhang, Xiaolin and Wei, Yunchao and Yang, Yi and Huang, Thomas S},
  journal={IEEE Transactions on Cybernetics},
  year={2020},
  publisher={IEEE}
}