Sssegmentation

<div align="center"> <img src="./docs/logo.png" width="600"/> </div> <br /> <p align="center"> <a href="https://sssegmentation.readthedocs.io/en/latest/"> <img src="https://img.shields.io/badge/docs-latest-blue" alt="Docs"> </a> <a href="https://pypi.org/project/sssegmentation/"> <img src="https://img.shields.io/pypi/pyversions/sssegmentation" alt="PyPI - Python Version"> </a> <a href="https://pypi.org/project/sssegmentation"> <img src="https://img.shields.io/pypi/v/sssegmentation" alt="PyPI"> </a> <a href="https://github.com/SegmentationBLWX/sssegmentation/blob/main/LICENSE"> <img src="https://img.shields.io/github/license/SegmentationBLWX/sssegmentation.svg" alt="License"> </a> <a href="https://pypi.org/project/sssegmentation/"> <img src="https://static.pepy.tech/badge/sssegmentation" alt="PyPI - Downloads"> </a> <a href="https://pypi.org/project/sssegmentation/"> <img src="https://static.pepy.tech/badge/sssegmentation/month" alt="PyPI - Downloads"> </a> <a href="https://github.com/SegmentationBLWX/modelstore"> <img src="https://img.shields.io/github/downloads/SegmentationBLWX/modelstore/total?style=flat-square" alt="Release - Downloads"> </a> <a href="https://github.com/SegmentationBLWX/sssegmentation/issues"> <img src="https://isitmaintained.com/badge/resolution/SegmentationBLWX/sssegmentation.svg" alt="Issue Resolution"> </a> <a href="https://github.com/SegmentationBLWX/sssegmentation/issues"> <img src="https://isitmaintained.com/badge/open/SegmentationBLWX/sssegmentation.svg" alt="Open Issues"> </a> <a href="https://github.com/SegmentationBLWX/sssegmentation/commits/main/"> <img src="https://img.shields.io/github/last-commit/SegmentationBLWX/sssegmentation/main.svg" alt="GitHub Last Commit (Main)"> </a> </p> <p align="center"> Documents: <a href="https://sssegmentation.readthedocs.io/en/latest/">https://sssegmentation.readthedocs.io/en/latest/</a> </p>

What's New

• 2024-08-05: Support SAMV2, refer to inference-with-samv2 for more details.
• 2023-12-20: Support EdgeSAM and SAMHQ, refer to inference-with-edgesam and inference-with-samhq for more details.
• 2023-10-25: Support ConvNeXtV2, refer to Results and Models for ConvNeXtV2 for more details.
• 2023-10-23: Support MobileViT and MobileViTV2, refer to Results and Models for MobileViT for more details.
• 2023-10-18: Support Mask2Former, refer to Results and Models for Mask2Former for more details.
• 2023-10-17: We release the source codes of IDRNet: Intervention-Driven Relation Network for Semantic Segmentation, which was accepted by NeurIPS 2023, refer to Results and Models for IDRNet for more details.
• 2023-10-15: Support MobileSAM, refer to inference-with-mobilesam for more details.
• 2023-09-27: Support SAM, refer to inference-with-sam for more details.

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

• High Performance

  The performance of re-implemented segmentation algorithms is better than or comparable to other codebases.

• Modular Design and Unified Benchmark

  Various segmentation methods are unified into several specific modules. Benefiting from this design, SSSegmentation can integrate a great deal of popular and contemporary semantic segmentation frameworks and then, train and test them on unified benchmarks.

• Fewer Dependencies

  SSSegmenation tries its best to avoid introducing more dependencies when reproducing novel semantic segmentation approaches.

Benchmark and Model Zoo

Supported Backbones

| Backbone | Model Zoo | Paper Link | Code Snippet | | :-: | :-: | :-: | :-: | | ConvNeXtV2 | Click | CVPR 2023 | Click | | MobileViTV2 | Click | ArXiv 2022 | Click | | ConvNeXt | Click | CVPR 2022 | Click | | MAE | Click | CVPR 2022 | Click | | MobileViT | Click | ICLR 2022 | Click | | BEiT | Click | ICLR 2022 | Click | | Twins | Click | NeurIPS 2021 | Click | | SwinTransformer | Click | ICCV 2021 | Click | | VisionTransformer | Click | IClR 2021 | Click | | BiSeNetV2 | Click | IJCV 2021 | Click | | ResNeSt | Click | ArXiv 2020 | Click | | CGNet | Click | TIP 2020 | Click | | HRNet | Click | CVPR 2019 | Click | | MobileNetV3 | Click | ICCV 2019 | Click | | FastSCNN | Click | ArXiv 2019 | Click | | BiSeNetV1 | Click | ECCV 2018 | Click | | MobileNetV2 | Click | CVPR 2018 | Click | | ERFNet | Click | T-ITS 2017 | Click | | ResNet | Click | CVPR 2016 | Click | | UNet | Click | MICCAI 2015 | Click |

Supported Segmentors

| Segmentor | Model Zoo | Paper Link | Code Snippet | | :-: | :-: | :-: | :-: | | SAMV2 | Click | ArXiv 2024 | [Click](./ssse