Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSeg
提交
50e2ca20
P
PaddleSeg
项目概览
PaddlePaddle
/
PaddleSeg
通知
286
Star
8
Fork
1
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
53
列表
看板
标记
里程碑
合并请求
3
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleSeg
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
53
Issue
53
列表
看板
标记
里程碑
合并请求
3
合并请求
3
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
50e2ca20
编写于
8月 26, 2019
作者:
W
wuzewu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix reference issue
上级
de101d65
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
22 addition
and
21 deletion
+22
-21
docs/configs/basic_group.md
docs/configs/basic_group.md
+5
-5
docs/data_prepare.md
docs/data_prepare.md
+3
-2
docs/model_zoo.md
docs/model_zoo.md
+11
-11
docs/usage.md
docs/usage.md
+3
-3
未找到文件。
docs/configs/basic_group.md
浏览文件 @
50e2ca20
...
...
@@ -4,7 +4,7 @@ BASIC Group存放所有通用配置
## `MEAN`
图像预处理减去的均值(格式为
*[R, G, B]*
)
图像预处理减去的均值(格式为
*[R, G, B]*
)
### 默认值
...
...
@@ -15,7 +15,7 @@ BASIC Group存放所有通用配置
## `STD`
图像预处理所除的标准差(格式为
*[R, G, B]*
)
图像预处理所除的标准差(格式为
*[R, G, B]*
)
### 默认值
...
...
@@ -26,7 +26,7 @@ BASIC Group存放所有通用配置
## `EVAL_CROP_SIZE`
评估时所对图片裁剪的大小(格式为
*[宽, 高]*
)
评估时所对图片裁剪的大小(格式为
*[宽, 高]*
)
### 默认值
...
...
@@ -40,7 +40,7 @@ BASIC Group存放所有通用配置
## `TRAIN_CROP_SIZE`
训练时所对图片裁剪的大小(格式为
*[宽, 高]*
)
训练时所对图片裁剪的大小(格式为
*[宽, 高]*
)
### 默认值
...
...
@@ -66,4 +66,4 @@ BASIC Group存放所有通用配置
*
增大BATCH_SIZE有利于模型训练时的收敛速度,但是会带来显存的开销。请根据实际情况评估后填写合适的值
<br/>
<br/>
\ No newline at end of file
<br/>
docs/data_prepare.md
浏览文件 @
50e2ca20
...
...
@@ -55,7 +55,8 @@ PaddleSeg采用通用的文件列表方式组织训练集、验证集和测试
*
文件列表请使用
**UTF-8**
格式保存, PaddleSeg默认使用UTF-8编码读取file_list文件
如下图所示,左边为原图的图片路径,右边为图片对应的标注路径。
![
cityscapes_filelist
](
./docs/imgs/file_list.png
)
![
cityscapes_filelist
](
./imgs/file_list.png
)
完整的配置信息可以参考
[
`./dataset/cityscapes_demo`
](
../dataset/cityscapes_demo/
)
目录下的yaml和文件列表。
...
...
@@ -104,4 +105,4 @@ python pdseg/check.py --cfg ${YAML_FILE_PATH}
-
当
`AUG.AUG_METHOD`
为rangscaling时,
`EVAL_CROP_SIZE`
的宽高应不小于缩放后图像中最大的宽高。
我们将计算并给出
`EVAL_CROP_SIZE`
的建议值。
\ No newline at end of file
我们将计算并给出
`EVAL_CROP_SIZE`
的建议值。
docs/model_zoo.md
浏览文件 @
50e2ca20
...
...
@@ -2,27 +2,27 @@
PaddleSeg对所有内置的分割模型都提供了公开数据集的下的预训练模型,通过加载预训练模型后训练可以在自定义数据集中得到更稳定地效果。
## ImageNet预训练模型
## ImageNet预训练模型
所有Imagenet预训练模型来自于PaddlePaddle图像分类库,想获取更多细节请点击
[
这里
](
https://github.com/PaddlePaddle/models/tree/develop/PaddleCV/image_classification
)
)
| 模型 | 数据集合 | Depth multiplier | 模型加载config设置 | 下载地址 | Accuray Top1/5 Error|
|---|---|---|---|---|---|
| MobieNetV2_1.0x | ImageNet | 1.0x | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: mobilenet
<br>
MODEL.DEEPLAB.DEPTH_MULTIPLIER: 1.0
<br>
MODEL.DEFAULT_NORM_TYPE: bn| [MobileNetV2_1.0x]
(https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV2_pretrained.tar) | 72.15%/90.65% |
| MobieNetV2_0.25x | ImageNet | 0.25x | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: mobilenet
<br>
MODEL.DEEPLAB.DEPTH_MULTIPLIER: 0.25
<br>
MODEL.DEFAULT_NORM_TYPE: bn |[MobileNetV2_0.25x]
(https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV2_x0_25_pretrained.tar) | 53.21%/76.52% |
| MobieNetV2_0.5x | ImageNet | 0.5x | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: mobilenet
<br>
MODEL.DEEPLAB.DEPTH_MULTIPLIER: 0.5
<br>
MODEL.DEFAULT_NORM_TYPE: bn | [MobileNetV2_0.5x]
(https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV2_x0_5_pretrained.tar) | 65.03%/85.72% |
| MobieNetV2_1.5x | ImageNet | 1.5x | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: mobilenet
<br>
MODEL.DEEPLAB.DEPTH_MULTIPLIER: 1.5
<br>
MODEL.DEFAULT_NORM_TYPE: bn| [MobileNetV2_1.5x]
(https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV2_x1_5_pretrained.tar) | 74.12%/91.67% |
| MobieNetV2_2.0x | ImageNet | 2.0x | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: mobilenet
<br>
MODEL.DEEPLAB.DEPTH_MULTIPLIER: 2.0
<br>
MODEL.DEFAULT_NORM_TYPE: bn | [MobileNetV2_2.0x]
(https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV2_x2_0_pretrained.tar) | 75.23%/92.58% |
| MobieNetV2_1.0x | ImageNet | 1.0x | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: mobilenet
<br>
MODEL.DEEPLAB.DEPTH_MULTIPLIER: 1.0
<br>
MODEL.DEFAULT_NORM_TYPE: bn|
[
MobileNetV2_1.0x
](
https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV2_pretrained.tar
)
| 72.15%/90.65% |
| MobieNetV2_0.25x | ImageNet | 0.25x | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: mobilenet
<br>
MODEL.DEEPLAB.DEPTH_MULTIPLIER: 0.25
<br>
MODEL.DEFAULT_NORM_TYPE: bn |
[
MobileNetV2_0.25x
](
https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV2_x0_25_pretrained.tar
)
| 53.21%/76.52% |
| MobieNetV2_0.5x | ImageNet | 0.5x | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: mobilenet
<br>
MODEL.DEEPLAB.DEPTH_MULTIPLIER: 0.5
<br>
MODEL.DEFAULT_NORM_TYPE: bn |
[
MobileNetV2_0.5x
](
https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV2_x0_5_pretrained.tar
)
| 65.03%/85.72% |
| MobieNetV2_1.5x | ImageNet | 1.5x | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: mobilenet
<br>
MODEL.DEEPLAB.DEPTH_MULTIPLIER: 1.5
<br>
MODEL.DEFAULT_NORM_TYPE: bn|
[
MobileNetV2_1.5x
](
https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV2_x1_5_pretrained.tar
)
| 74.12%/91.67% |
| MobieNetV2_2.0x | ImageNet | 2.0x | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: mobilenet
<br>
MODEL.DEEPLAB.DEPTH_MULTIPLIER: 2.0
<br>
MODEL.DEFAULT_NORM_TYPE: bn |
[
MobileNetV2_2.0x
](
https://paddle-imagenet-models-name.bj.bcebos.com/MobileNetV2_x2_0_pretrained.tar
)
| 75.23%/92.58% |
用户可以结合实际场景的精度和预测性能要求,选取不同
`Depth multiplier`
参数的MobileNet模型。
| 模型 | 数据集合 | 模型加载config设置 | 下载地址 | Accuray Top1/5 Error |
|---|---|---|---|---|
| Xception41 | ImageNet | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: xception_41
<br>
MODEL.DEFAULT_NORM_TYPE: bn| [Xception41_pretrained.tgz]
(https://paddleseg.bj.bcebos.com/models/Xception41_pretrained.tgz) | 79.5%/94.38% |
| Xception65 | ImageNet | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: xception_65
<br>
MODEL.DEFAULT_NORM_TYPE: bn| [Xception65_pretrained.tgz]
(https://paddleseg.bj.bcebos.com/models/Xception65_pretrained.tgz) | 80.32%/94.47% |
| Xception41 | ImageNet | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: xception_41
<br>
MODEL.DEFAULT_NORM_TYPE: bn|
[
Xception41_pretrained.tgz
](
https://paddleseg.bj.bcebos.com/models/Xception41_pretrained.tgz
)
| 79.5%/94.38% |
| Xception65 | ImageNet | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: xception_65
<br>
MODEL.DEFAULT_NORM_TYPE: bn|
[
Xception65_pretrained.tgz
](
https://paddleseg.bj.bcebos.com/models/Xception65_pretrained.tgz
)
| 80.32%/94.47% |
| Xception71 | ImageNet | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: xception_71
<br>
MODEL.DEFAULT_NORM_TYPE: bn| coming soon | -- |
## COCO预训练模型
## COCO预训练模型
train数据集为coco instance分割数据集合转换成的语义分割数据集合
...
...
@@ -32,13 +32,13 @@ train数据集为coco instance分割数据集合转换成的语义分割数据
| DeeplabV3+/Xception65/bn | COCO | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: xception_65
<br>
MODEL.DEFAULT_NORM_TYPE: bn |
[
xception65_coco.tgz
](
https://paddleseg.bj.bcebos.com/models/xception65_coco.tgz
)
| 16 | -- | -- |
| UNet/bn | COCO | MODEL.MODEL_NEME: unet
<br>
MODEL.DEFAULT_NORM_TYPE: bn |
[
unet
](
https://paddleseg.bj.bcebos.com/models/unet_coco_v2.tgz
)
| 16 | -- | -- |
## Cityscapes预训练模型
## Cityscapes预训练模型
train数据集合为Cityscapes 训练集合,测试为Cityscapes的验证集合
| 模型 | 数据集合 | 模型加载config设置 | 下载地址 |Output Stride| mutli-scale test| mIoU on val|
|---|---|---|---|---|---|---|
| DeepLabv3+/MobileNetv2/bn | Cityscapes |MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: mobilenet
<br>
MODEL.DEEPLAB.DEPTH_MULTIPLIER: 1.0
<br>
MODEL.DEEPLAB.ENCODER_WITH_ASPP: False
<br>
MODEL.DEEPLAB.ENABLE_DECODER: False
<br>
MODEL.DEFAULT_NORM_TYPE: bn|[mobilenet_cityscapes.tgz]
(https://paddleseg.bj.bcebos.com/models/mobilenet_cityscapes.tgz) |16|false| 0.698|
| DeepLabv3+/MobileNetv2/bn | Cityscapes |MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: mobilenet
<br>
MODEL.DEEPLAB.DEPTH_MULTIPLIER: 1.0
<br>
MODEL.DEEPLAB.ENCODER_WITH_ASPP: False
<br>
MODEL.DEEPLAB.ENABLE_DECODER: False
<br>
MODEL.DEFAULT_NORM_TYPE: bn|
[
mobilenet_cityscapes.tgz
](
https://paddleseg.bj.bcebos.com/models/mobilenet_cityscapes.tgz
)
|16|false| 0.698|
| DeepLabv3+/Xception65/gn | Cityscapes |MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: xception_65
<br>
MODEL.DEFAULT_NORM_TYPE: gn |
[
deeplabv3p_xception65_cityscapes.tgz
](
https://paddleseg.bj.bcebos.com/models/deeplabv3p_xception65_cityscapes.tgz
)
|16|false| 0.7804 |
| DeepLabv3+/Xception65/bn | Cityscapes | MODEL.MODEL_NAME: deeplabv3p
<br>
MODEL.DEEPLAB.BACKBONE: xception_65
<br>
MODEL.DEFAULT_NORM_TYPE: bn|
[
Xception65_deeplab_cityscapes.tgz
](
https://paddleseg.bj.bcebos.com/models/Xception65_deeplab_cityscapes.tgz
)
| 16 | false | 0.7715 |
| ICNet/bn | Cityscapes | MODEL.MODEL_NAME: icnet
<br>
MODEL.DEFAULT_NORM_TYPE: bn |
[
icnet_cityscapes.tgz
](
https://paddleseg.bj.bcebos.com/models/icnet_cityscapes.tgz
)
|16|false| 0.6854 |
docs/usage.md
浏览文件 @
50e2ca20
...
...
@@ -48,7 +48,7 @@ python pdseg/export_model.py ${FLAGS} ${OPTIONS}
1.
下载合适版本的paddlepaddle
2.
PaddleSeg相关依赖已经安装
如果有不确认的地方,请参考
[
安装说明
](
./
docs/
installation.md
)
如果有不确认的地方,请参考
[
安装说明
](
./installation.md
)
### 下载预训练模型
```
shell
...
...
@@ -111,10 +111,10 @@ NOTE:
启动TensorBoard命令后,我们可以在浏览器中查看对应的训练数据
在
`SCALAR`
这个tab中,查看训练loss、iou、acc的变化趋势
![](
docs
/imgs/tensorboard_scalar.JPG
)
![](
.
/imgs/tensorboard_scalar.JPG
)
在
`IMAGE`
这个tab中,查看样本的预测情况
![](
docs
/imgs/tensorboard_image.JPG
)
![](
.
/imgs/tensorboard_image.JPG
)
### 模型评估
训练完成后,我们可以通过eval.py来评估模型效果。由于我们设置的训练EPOCH数量为500,保存间隔为10,因此一共会产生50个定期保存的模型,加上最终保存的final模型,一共有51个模型。我们选择最后保存的模型进行效果的评估:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录