Install and Citations

Install from Source

  • Install PyTorch by following the PyTorch instructions. This package relies on PyTorch master branch (higher than stable released v0.4.0), please follow the instruction to install PyTorch from source.

  • PIP Install:

    pip install torch-encoding
    
  • Install from source:

    git clone https://github.com/zhanghang1989/PyTorch-Encoding && cd PyTorch-Encoding
    python setup.py install
    

Citations

Note

If using the code in your research, please cite our papers.

  • Hang Zhang, Kristin Dana, Jianping Shi, Zhongyue Zhang, Xiaogang Wang, Ambrish Tyagi, Amit Agrawal. “Context Encoding for Semantic Segmentation” The IEEE Conference on Computer Vision and Pattern Recognition (CVPR) 2018:

    @InProceedings{Zhang_2018_CVPR,
    author = {Zhang, Hang and Dana, Kristin and Shi, Jianping and Zhang, Zhongyue and Wang, Xiaogang and Tyagi, Ambrish and Agrawal, Amit},
    title = {Context Encoding for Semantic Segmentation},
    booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
    month = {June},
    year = {2018}
    }
    
  • Hang Zhang, Jia Xue, and Kristin Dana. “Deep TEN: Texture Encoding Network.” The IEEE Conference on Computer Vision and Pattern Recognition (CVPR) 2017:

    @InProceedings{Zhang_2017_CVPR,
    author = {Zhang, Hang and Xue, Jia and Dana, Kristin},
    title = {Deep TEN: Texture Encoding Network},
    booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
    month = {July},
    year = {2017}
    }