Gzzgz/sssegmentation

Documents: https://sssegmentation.readthedocs.io/en/latest/

Introduction

SSSegmentation is an open source supervised semantic segmentation toolbox based on PyTorch. You can star this repository to keep track of the project if it's helpful for you, thank you for your support.

Major Features

Unified Benchmark

We provide a unified benchmark toolbox for various semantic segmentation methods.
Modular Design

We decompose the semantic segmentation framework into different components and one can easily construct a customized semantic segmentation framework by combining different modules.
Support of Multiple Methods Out of Box

The toolbox directly supports popular and contemporary semantic segmentation frameworks, e.g., ISNet, DeepLabV3, PSPNet, MCIBI, etc.
High Performance

The segmentation performance is better than or comparable to other codebases.

Benchmark and Model Zoo

Supported Backbones

Supported Segmentors

Supported Datasets

Citation

If you use this framework in your research, please cite this project:

@misc{ssseg2020,
    author = {Zhenchao Jin},
    title = {SSSegmentation: An Open Source Supervised Semantic Segmentation Toolbox Based on PyTorch},
    year = {2020},
    publisher = {GitHub},
    journal = {GitHub repository},
    howpublished = {\url{https://github.com/SegmentationBLWX/sssegmentation}},
}

@inproceedings{jin2021isnet,
    title={ISNet: Integrate Image-Level and Semantic-Level Context for Semantic Segmentation},
    author={Jin, Zhenchao and Liu, Bin and Chu, Qi and Yu, Nenghai},
    booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
    pages={7189--7198},
    year={2021}
}

@inproceedings{jin2021mining,
    title={Mining Contextual Information Beyond Image for Semantic Segmentation},
    author={Jin, Zhenchao and Gong, Tao and Yu, Dongdong and Chu, Qi and Wang, Jian and Wang, Changhu and Shao, Jie},
    booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
    pages={7231--7241},
    year={2021}
}

References

About

SSSegmentation: An Open Source Supervised Semantic Segmentation Toolbox Based on PyTorch.

https://sssegmentation.readthedocs.io/en/latest/

Apache License 2.0

Languages

Language:Python 100.0%