hustvl / DGCN

Deep Graph Cut Network for Weakly-supervised Semantic Segmentation.

Jiapei Feng, Xinggang Wang, Wenyu Liu

in SCIENCE CHINA Information Sciences (SCIS), Volume 64 , Issue 3 : 130105(2021) https://doi.org/10.1007/s11432-020-3065-4

Getting Started

To get a local copy up and running follow these simple steps.

Prerequisites

• You need to install commonly-used libraries such as Numpy and Pytorch.
• The CAM seed file you can find in DSRG.

Train

1. Run.

   bash run.sh

License

Distributed under the MIT License. See LICENSE for more information.

Acknowledgements

This code is heavily borrowed from pytorch-deeplab-resnet.

Citation

If you find this code is useful, please cite it using the following bibtex information.

@article{feng2021deep,
  title={Deep Graph Cut Network for Weakly-supervised Semantic Segmentation},
  author={Feng, Jiapei and Wang, Xinggang and Liu Wenyu},
  journal={Science China Information Sciences},
  volume={64},
  number={3},
  pages={1--13},
  year={2021},
  publisher={Science China Press}
}