Skip to content

Easy-to-use image segmentation library with awesome pre-trained model zoo, supporting wide-range of practical tasks in Semantic Segmentation, Interactive Segmentation, Panoptic Segmentation, Image Matting, 3D Segmentation, etc.

License

Notifications You must be signed in to change notification settings

PaddlePaddle/PaddleSeg

Folders and files

NameName
Last commit message
Last commit date

Latest commit

2eddffc · Dec 14, 2020
Dec 7, 2020
Dec 14, 2020
Dec 7, 2020
Dec 7, 2020
Dec 10, 2020
Dec 14, 2020
Dec 7, 2020
Aug 26, 2019
Jun 5, 2020
Aug 26, 2019
Aug 26, 2019
May 15, 2020
Aug 26, 2019
Dec 14, 2020
Dec 14, 2020
Dec 7, 2020
Dec 7, 2020
Dec 7, 2020
Dec 7, 2020

Repository files navigation

English | 简体中文

PaddleSeg(Dynamic Graph)

Build Status License Version python version support os

Welcome to the dynamic version! PaddleSeg is the first development kit which supports PaddlePaddle 2.0. Currently, we provide an experimental version that allows developers to have full-featured experience on dynamic graph. In the near future, the dynamic version will be set as default, and the static one will be moved to "legacy" directory.

The full-detailed documents and tutorials are coming soon. So far there are minimum tutorials that help you to enjoy the strengths of dynamic version.

Model Zoo

Model\Backbone ResNet50 ResNet101 HRNetw18 HRNetw48
ANN
BiSeNetv2 - - - -
DANet
Deeplabv3
Deeplabv3P
Fast-SCNN - - - -
FCN
GCNet
GSCNN
OCRNet
PSPNet
U-Net - - - -
U2-Net - - - -
Att U-Net - - - -

Dataset

  • Cityscapes
  • Pascal VOC
  • ADE20K
  • Pascal Context
  • COCO stuff

Installation

  1. Install PaddlePaddle

System Requirements:

  • PaddlePaddle >= 2.0.0rc
  • Python >= 3.6+

Note: the above requirements are for the dynamic graph version. If you intent to use the static one, please refers to here.

Highly recommend you install the GPU version of PaddlePaddle, due to large overhead of segmentation models, otherwise it could be out of memory while running the models.

For more detailed installation tutorials, please refer to the official website of PaddlePaddle

Download PaddleSeg

git clone https://github.com/PaddlePaddle/PaddleSeg

Install Dependencies

Install the python dependencies via the following commands,and please make sure execute it at least once in your branch.

cd PaddleSeg
export PYTHONPATH=`pwd`
# Run the following one on Windows
# set PYTHONPATH=%cd%
pip install -r requirements.txt

Quick Training

python train.py --config configs/quick_start/bisenet_optic_disc_512x512_1k.yml

Tutorials

Feedbacks and Contact

  • The dynamic version is still under development, if you find any issue or have an idea on new features, please don't hesitate to contact us via GitHub Issues.
  • PaddleSeg User Group (QQ): 850378321 or 793114768

Acknowledgement

  • Thanks jm12138 for contributing U2-Net.
  • Thanks zjhellofss (Fu Shenshen) for contributing Attention U-Net.