Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSeg
提交
543dd97e
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看板
提交
543dd97e
编写于
5月 19, 2020
作者:
C
chenguowei01
浏览文件
操作
浏览文件
下载
差异文件
Merge branch 'develop' of
https://github.com/PaddlePaddle/PaddleSeg
into humanseg
上级
d9af2488
4f33a23c
变更
8
隐藏空白更改
内联
并排
Showing
8 changed file
with
20 addition
and
16 deletion
+20
-16
README.md
README.md
+1
-1
contrib/HumanSeg/README.md
contrib/HumanSeg/README.md
+4
-0
contrib/HumanSeg/requirements.txt
contrib/HumanSeg/requirements.txt
+1
-1
contrib/LaneNet/requirements.txt
contrib/LaneNet/requirements.txt
+1
-1
contrib/README.md
contrib/README.md
+8
-8
contrib/RemoteSensing/requirements.txt
contrib/RemoteSensing/requirements.txt
+1
-1
docs/model_export.md
docs/model_export.md
+3
-3
requirements.txt
requirements.txt
+1
-1
未找到文件。
README.md
浏览文件 @
543dd97e
...
...
@@ -121,7 +121,7 @@ pip install -r requirements.txt
|U-Net图像分割|
[
点击体验
](
https://aistudio.baidu.com/aistudio/projectDetail/102889
)
|
|DeepLabv3+图像分割|
[
点击体验
](
https://aistudio.baidu.com/aistudio/projectDetail/226703
)
|
|工业质检(零件瑕疵检测)|
[
点击体验
](
https://aistudio.baidu.com/aistudio/projectdetail/184392
)
|
|人像分割|
[
点击体验
](
https://aistudio.baidu.com/aistudio/projectdetail/
188833
)
|
|人像分割|
[
点击体验
](
https://aistudio.baidu.com/aistudio/projectdetail/
475345
)
|
|PaddleSeg特色垂类模型|
[
点击体验
](
https://aistudio.baidu.com/aistudio/projectdetail/226710
)
|
## FAQ
...
...
contrib/HumanSeg/README.md
浏览文件 @
543dd97e
...
...
@@ -184,3 +184,7 @@ python quant_online.py --model_type HumanSegMobile \
*
`--learning_rate`
: 初始学习率
*
`--num_epochs`
: 训练轮数
*
`--image_shape`
: 网络输入图像大小(w, h)
## AIStudio在线教程
我们在AI Studio平台上提供了人像分割在线体验的教程,
[
点击体验
](
https://aistudio.baidu.com/aistudio/projectdetail/475345
)
contrib/HumanSeg/requirements.txt
浏览文件 @
543dd97e
visualdl == 2.0.0
-alpha.
1
visualdl == 2.0.0
b
1
paddleslim
contrib/LaneNet/requirements.txt
浏览文件 @
543dd97e
...
...
@@ -9,4 +9,4 @@ opencv-python
tqdm
requests
sklearn
visualdl == 2.0.0
-alpha.2
visualdl == 2.0.0
b1
contrib/README.md
浏览文件 @
543dd97e
...
...
@@ -40,7 +40,7 @@ PaddleSeg提供了基于LaneNet的车道线分割模型,更多详情请点击[
U-Net
### 2. 数据准备
执行以下命令下载并解压数据集,数据集将存放在contrib/MechanicalIndustryMeter文件夹下:
```
...
...
@@ -56,8 +56,8 @@ python ./pretrained_model/download_model.py unet_bn_coco
### 4. 训练与评估
```
export CUDA_VISIBLE_DEVICES=0
python ./pdseg/train.py --log_steps 10 --cfg contrib/MechanicalIndustryMeter/unet_mechanical_meter.yaml --use_gpu --do_eval --use_mpio
export CUDA_VISIBLE_DEVICES=0
python ./pdseg/train.py --log_steps 10 --cfg contrib/MechanicalIndustryMeter/unet_mechanical_meter.yaml --use_gpu --do_eval --use_mpio
```
### 5. 可视化
...
...
@@ -71,18 +71,18 @@ python ./contrib/MechanicalIndustryMeter/download_unet_mechanical_industry_meter
```
python ./pdseg/vis.py --cfg contrib/MechanicalIndustryMeter/unet_mechanical_meter.yaml --use_gpu --vis_dir vis_meter \
TEST.TEST_MODEL "./contrib/MechanicalIndustryMeter/unet_mechanical_industry_meter/"
TEST.TEST_MODEL "./contrib/MechanicalIndustryMeter/unet_mechanical_industry_meter/"
```
可视化结果会保存在./vis_meter文件夹下。
### 6. 可视化结果示例:
原图:
!
[](
MechanicalIndustryMeter/imgs/1560143028.5_IMG_3091.JPG
)
预测结果:
!
[](
MechanicalIndustryMeter/imgs/1560143028.5_IMG_3091.png
)
## AIStudio在线教程
...
...
@@ -92,5 +92,5 @@ PaddleSeg在AI Studio平台上提供了在线体验的教程,欢迎体验:
|教程|链接|
|-|-|
|工业质检|
[
点击体验
](
https://aistudio.baidu.com/aistudio/projectdetail/184392
)
|
|人像分割|
[
点击体验
](
https://aistudio.baidu.com/aistudio/projectdetail/
188833
)
|
|人像分割|
[
点击体验
](
https://aistudio.baidu.com/aistudio/projectdetail/
475345
)
|
|特色垂类模型|
[
点击体验
](
https://aistudio.baidu.com/aistudio/projectdetail/226710
)
|
contrib/RemoteSensing/requirements.txt
浏览文件 @
543dd97e
visualdl
>= 2.0.0-alpha.2
visualdl
== 2.0.0b1
docs/model_export.md
浏览文件 @
543dd97e
...
...
@@ -2,7 +2,7 @@
通过训练得到一个满足要求的模型后,如果想要将该模型接入到C++预测库或者Serving服务,我们需要通过
`pdseg/export_model.py`
来导出该模型。
该脚本的使用方法和
`train.py/eval.py/vis.py`
完全一样
该脚本的使用方法和
`train.py/eval.py/vis.py`
完全一样
。
## FLAGS
...
...
@@ -12,10 +12,10 @@
## 使用示例
我们使用
[
训练/评估/可视化
](
./usage.md
)
一节中训练得到的模型进行试用,
脚本
如下
我们使用
[
训练/评估/可视化
](
./usage.md
)
一节中训练得到的模型进行试用,
命令
如下
```
shell
python pdseg/export_model.py
--cfg
configs/unet_
pet.yaml TEST.TEST_MODEL
test
/saved_models/unet_pet
/final
python pdseg/export_model.py
--cfg
configs/unet_
optic.yaml TEST.TEST_MODEL ./saved_model/unet_optic
/final
```
预测模型会导出到
`freeze_model`
目录,用于
`C++`
或者
`Python`
预测的模型配置会导出到
`freeze_model/deploy.yaml`
下
requirements.txt
浏览文件 @
543dd97e
...
...
@@ -2,4 +2,4 @@ pre-commit
yapf
== 0.26.0
flake8
pyyaml
>= 5.1
visualdl
== 2.0.0
-alpha.2
visualdl
== 2.0.0
b1
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录