/mmsegmentation-mpcnet

Primary LanguagePythonApache License 2.0Apache-2.0

 
OpenMMLab website HOT      OpenMMLab platform TRY IT OUT
 

PyPI - Python Version PyPI docs badge codecov license issue resolution open issues

Documentation: https://mmsegmentation.readthedocs.io/

English | 简体中文

Introduction

This is the implementation for Multi-scale Prototype Contrast Network.

demo image

The trained model for iSAID can be found at the BaiduYun (password: cbn0), corresponding config is configs/mpcnet/mpc_swin_isaid.py.

Major features

  • Unified Benchmark

    We provide a unified benchmark toolbox for various semantic segmentation methods. This code is developed based on mmsegemetnation.

  • 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.

Datasets preparation:

Installation

Please refer to get_started.md for installation and dataset_prepare.md for dataset preparation.

Get Started

Please see train.md and inference.md for the basic usage of MMSegmentation. There are also tutorials for customizing dataset, designing data pipeline, customizing modules, and customizing runtime. We also provide many training tricks for better training and useful tools for deployment.

A Colab tutorial is also provided. You may preview the notebook here or directly run on Colab.

Please refer to FAQ for frequently asked questions.

Citation

If you find this project useful in your research, please consider cite:

@misc{mmseg2020,
    title={{MMSegmentation}: OpenMMLab Semantic Segmentation Toolbox and Benchmark},
    author={MMSegmentation Contributors},
    howpublished = {\url{https://github.com/open-mmlab/mmsegmentation}},
    year={2020}

}

Contributing

We appreciate all contributions to improve MMSegmentation. Please refer to CONTRIBUTING.md for the contributing guideline.

Acknowledgement

MMSegmentation is an open source project that welcome any contribution and feedback. We wish that the toolbox and benchmark could serve the growing research community by providing a flexible as well as standardized toolkit to reimplement existing methods and develop their own new semantic segmentation methods.