Skip to content

Latest commit

 

History

History
71 lines (50 loc) · 1.91 KB

README.md

File metadata and controls

71 lines (50 loc) · 1.91 KB

PaddleSeg(动态图版本)

本目录提供了PaddleSeg的动态图版本,目前已经完成了模型训练、评估、数据处理等功能,在未来的版本中,PaddleSeg将会启动默认的动态图模式。目前该目录处于实验阶段,如果您在使用过程中遇到任何问题,请通过issue反馈给我们,我们将会在第一时间跟进处理。

模型库

模型\骨干网络 ResNet50 ResNet101 HRNetw18 HRNetw48
ANN
BiSeNetv2 - - - -
DANet
Deeplabv3
Deeplabv3P
Fast-SCNN - - - -
FCN
GCNet
OCRNet
PSPNet
UNet - - - -

数据集

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

安装

  1. 安装PaddlePaddle

版本要求

  • PaddlePaddle >= 2.0.0rc

  • Python >= 3.6+

由于图像分割模型计算开销大,推荐在GPU版本的PaddlePaddle下使用PaddleSeg。推荐安装10.0以上的CUDA环境。

安装教程请见PaddlePaddle官网

  1. 下载PaddleSeg代码
git clone https://github.com/PaddlePaddle/PaddleSeg
  1. 安装PaddleSeg依赖 通过以下命令安装python包依赖,请确保在该分支上至少执行过一次以下命令:
cd PaddleSeg/dygraph
export PYTHONPATH=`pwd`
# windows下请执行以下命令
# set PYTHONPATH=%cd%
pip install -r requirements.txt

训练

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

使用教程