Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSeg
提交
3f2a7a7a
P
PaddleSeg
项目概览
PaddlePaddle
/
PaddleSeg
通知
285
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看板
未验证
提交
3f2a7a7a
编写于
2月 26, 2020
作者:
L
LielinJiang
提交者:
GitHub
2月 26, 2020
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #170 from LielinJiang/release/v0.4.0
release change log
上级
bbd53e5c
7c4d39da
变更
8
隐藏空白更改
内联
并排
Showing
8 changed file
with
26 addition
and
7 deletion
+26
-7
README.md
README.md
+10
-2
contrib/LaneNet/README.md
contrib/LaneNet/README.md
+1
-1
docs/imgs/fast-scnn.png
docs/imgs/fast-scnn.png
+0
-0
docs/models.md
docs/models.md
+12
-0
slim/distillation/README.md
slim/distillation/README.md
+0
-1
slim/nas/README.md
slim/nas/README.md
+1
-1
slim/prune/README.md
slim/prune/README.md
+1
-1
turtorial/finetune_fast_scnn.md
turtorial/finetune_fast_scnn.md
+1
-1
未找到文件。
README.md
浏览文件 @
3f2a7a7a
...
@@ -6,7 +6,7 @@
...
@@ -6,7 +6,7 @@
## 简介
## 简介
PaddleSeg是基于
[
PaddlePaddle
](
https://www.paddlepaddle.org.cn
)
开发的语义分割库,覆盖了DeepLabv3+, U-Net, ICNet, PSPNet, HRNet等主流分割模型。通过统一的配置,帮助用户更便捷地完成从训练到部署的全流程图像分割应用。
PaddleSeg是基于
[
PaddlePaddle
](
https://www.paddlepaddle.org.cn
)
开发的语义分割库,覆盖了DeepLabv3+, U-Net, ICNet, PSPNet, HRNet
, Fast-SCNN
等主流分割模型。通过统一的配置,帮助用户更便捷地完成从训练到部署的全流程图像分割应用。
</br>
</br>
...
@@ -33,7 +33,7 @@ PaddleSeg是基于[PaddlePaddle](https://www.paddlepaddle.org.cn)开发的语义
...
@@ -33,7 +33,7 @@ PaddleSeg是基于[PaddlePaddle](https://www.paddlepaddle.org.cn)开发的语义
-
**模块化设计**
-
**模块化设计**
支持U-Net, DeepLabv3+, ICNet, PSPNet, HRNet
五
种主流分割网络,结合预训练模型和可调节的骨干网络,满足不同性能和精度的要求;选择不同的损失函数如Dice Loss, BCE Loss等方式可以强化小目标和不均衡样本场景下的分割精度。
支持U-Net, DeepLabv3+, ICNet, PSPNet, HRNet
, Fast-SCNN六
种主流分割网络,结合预训练模型和可调节的骨干网络,满足不同性能和精度的要求;选择不同的损失函数如Dice Loss, BCE Loss等方式可以强化小目标和不均衡样本场景下的分割精度。
-
**高性能**
-
**高性能**
...
@@ -163,6 +163,14 @@ A: 请将PaddlePaddle升级至1.5.2版本或以上。
...
@@ -163,6 +163,14 @@ A: 请将PaddlePaddle升级至1.5.2版本或以上。
<p
align=
"center"
>
   
微信公众号
                
官方技术交流QQ群
</p>
<p
align=
"center"
>
   
微信公众号
                
官方技术交流QQ群
</p>
## 更新日志
## 更新日志
*
2020.02.25
**`v0.4.0`**
*
新增适用于实时场景且不需要预训练模型的分割网络Fast-SCNN,提供基于Cityscapes的
[
预训练模型
](
./docs/model_zoo.md
)
1个。
*
新增LaneNet车道线检测网络,提供
[
预训练模型
](
https://github.com/PaddlePaddle/PaddleSeg/tree/release/v0.4.0/contrib/LaneNet#%E4%B8%83-%E5%8F%AF%E8%A7%86%E5%8C%96
)
一个。
*
新增基于PaddleSlim的分割库压缩策略(
[
量化
](
./slim/quantization/README.md
)
,
[
蒸馏
](
./slim/distillation/README.md
)
,
[
剪枝
](
./slim/prune/README.md
)
,
[
搜索
](
./slim/nas/README.md
)
)
*
2019.12.15
*
2019.12.15
**`v0.3.0`**
**`v0.3.0`**
...
...
contrib/LaneNet/README.md
浏览文件 @
3f2a7a7a
...
@@ -108,7 +108,7 @@ SOLVER:
...
@@ -108,7 +108,7 @@ SOLVER:
使用下述命令启动训练
使用下述命令启动训练
```
shell
```
shell
CUDA_VISIBLE_DEVICES
=
0 python
-u
train.py
--cfg
configs/lanenet.yaml
--use_gpu
--
use_mpio
--
do_eval
CUDA_VISIBLE_DEVICES
=
0 python
-u
train.py
--cfg
configs/lanenet.yaml
--use_gpu
--do_eval
```
```
## 六. 进行评估
## 六. 进行评估
...
...
docs/imgs/fast-scnn.png
0 → 100644
浏览文件 @
3f2a7a7a
117.6 KB
docs/models.md
浏览文件 @
3f2a7a7a
...
@@ -5,6 +5,7 @@
...
@@ -5,6 +5,7 @@
-
[
PSPNet
](
#PSPNet
)
-
[
PSPNet
](
#PSPNet
)
-
[
ICNet
](
#ICNet
)
-
[
ICNet
](
#ICNet
)
-
[
HRNet
](
#HRNet
)
-
[
HRNet
](
#HRNet
)
-
[
Fast-SCNN
](
#Fast-SCNN
)
## U-Net
## U-Net
U-Net [1] 起源于医疗图像分割,整个网络是标准的encoder-decoder网络,特点是参数少,计算快,应用性强,对于一般场景适应度很高。U-Net最早于2015年提出,并在ISBI 2015 Cell Tracking Challenge取得了第一。经过发展,目前有多个变形和应用。
U-Net [1] 起源于医疗图像分割,整个网络是标准的encoder-decoder网络,特点是参数少,计算快,应用性强,对于一般场景适应度很高。U-Net最早于2015年提出,并在ISBI 2015 Cell Tracking Challenge取得了第一。经过发展,目前有多个变形和应用。
...
@@ -58,6 +59,14 @@ HRNet在人体姿态估计、语义分割和目标检测领域都取得了显著
...
@@ -58,6 +59,14 @@ HRNet在人体姿态估计、语义分割和目标检测领域都取得了显著
![](
./imgs/hrnet.png
)
![](
./imgs/hrnet.png
)
### Fast-SCNN
Fast-SCNN [7] 是一个面向实时的语义分割网络。在双分支的结构基础上,大量使用了深度可分离卷积和逆残差(inverted-residual)模块,并且使用特征融合构造金字塔池化模块 (Pyramid Pooling Module)来融合上下文信息。这使得Fast-SCNN在保持高效的情况下能学习到丰富的细节信息。
整个网络结构如下:
![](
./imgs/fast-scnn.png
)
## 参考文献
## 参考文献
[
1] [U-Net: Convolutional Networks for Biomedical Image Segmentation
](
https://arxiv.org/abs/1505.04597
)
[
1] [U-Net: Convolutional Networks for Biomedical Image Segmentation
](
https://arxiv.org/abs/1505.04597
)
...
@@ -72,3 +81,6 @@ HRNet在人体姿态估计、语义分割和目标检测领域都取得了显著
...
@@ -72,3 +81,6 @@ HRNet在人体姿态估计、语义分割和目标检测领域都取得了显著
[
6] [Deep High-Resolution Representation Learning for Visual Recognition
](
https://arxiv.org/abs/1908.07919
)
[
6] [Deep High-Resolution Representation Learning for Visual Recognition
](
https://arxiv.org/abs/1908.07919
)
[
7] [Fast-SCNN: Fast Semantic Segmentation Network
](
https://arxiv.org/abs/1902.04502
)
slim/distillation/README.md
浏览文件 @
3f2a7a7a
...
@@ -89,7 +89,6 @@ python -m paddle.distributed.launch ./slim/distillation/train_distill.py \
...
@@ -89,7 +89,6 @@ python -m paddle.distributed.launch ./slim/distillation/train_distill.py \
--log_steps
10
--cfg
./slim/distillation/cityscape.yaml
\
--log_steps
10
--cfg
./slim/distillation/cityscape.yaml
\
--teacher_cfg
./slim/distillation/cityscape_teacher.yaml
\
--teacher_cfg
./slim/distillation/cityscape_teacher.yaml
\
--use_gpu
\
--use_gpu
\
--use_mpio
\
--do_eval
--do_eval
```
```
...
...
slim/nas/README.md
浏览文件 @
3f2a7a7a
...
@@ -46,7 +46,7 @@ SLIM:
...
@@ -46,7 +46,7 @@ SLIM:
## 训练与评估
## 训练与评估
执行以下命令,边训练边评估
执行以下命令,边训练边评估
```
shell
```
shell
CUDA_VISIBLE_DEVICES
=
0 python
-u
./slim/nas/train_nas.py
--log_steps
10
--cfg
configs/deeplabv3p_mobilenetv2_cityscapes.yaml
--use_gpu
--use_mpio
\
CUDA_VISIBLE_DEVICES
=
0 python
-u
./slim/nas/train_nas.py
--log_steps
10
--cfg
configs/deeplabv3p_mobilenetv2_cityscapes.yaml
--use_gpu
\
SLIM.NAS_PORT 23333
\
SLIM.NAS_PORT 23333
\
SLIM.NAS_ADDRESS
""
\
SLIM.NAS_ADDRESS
""
\
SLIM.NAS_SEARCH_STEPS 2
\
SLIM.NAS_SEARCH_STEPS 2
\
...
...
slim/prune/README.md
浏览文件 @
3f2a7a7a
...
@@ -46,7 +46,7 @@ SLIM.PRUNE_RATIOS '[0.1,0.1,0.1]'
...
@@ -46,7 +46,7 @@ SLIM.PRUNE_RATIOS '[0.1,0.1,0.1]'
```
shell
```
shell
CUDA_VISIBLE_DEVICES
=
0
CUDA_VISIBLE_DEVICES
=
0
python
-u
./slim/prune/eval_prune.py
--cfg
configs/cityscape_fast_scnn.yaml
--use_gpu
--use_mpio
\
python
-u
./slim/prune/eval_prune.py
--cfg
configs/cityscape_fast_scnn.yaml
--use_gpu
\
TEST.TEST_MODEL your_trained_model
\
TEST.TEST_MODEL your_trained_model
\
```
```
...
...
turtorial/finetune_fast_scnn.md
浏览文件 @
3f2a7a7a
...
@@ -114,6 +114,6 @@ python pdseg/eval.py --use_gpu --cfg ./configs/fast_scnn_pet.yaml
...
@@ -114,6 +114,6 @@ python pdseg/eval.py --use_gpu --cfg ./configs/fast_scnn_pet.yaml
| ICNet/bn |(1024, 2048) |8.76ms| 0.6831 |
| ICNet/bn |(1024, 2048) |8.76ms| 0.6831 |
| Fast-SCNN/bn | (1024, 2048) |6.28ms| 0.6964 |
| Fast-SCNN/bn | (1024, 2048) |6.28ms| 0.6964 |
上述测试环境为v100. 测试使用paddle的推理接口
[
zero_copy
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/advanced_
usage/deploy
/inference/python_infer_cn.html#id8
)
的方式,模型输出是类别,即argmax后的值。
上述测试环境为v100. 测试使用paddle的推理接口
[
zero_copy
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/advanced_
guide/inference_deployment
/inference/python_infer_cn.html#id8
)
的方式,模型输出是类别,即argmax后的值。
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录