Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
s920243400
PaddleDetection
提交
80b02773
P
PaddleDetection
项目概览
s920243400
/
PaddleDetection
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleDetection
通知
2
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleDetection
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
80b02773
编写于
4月 15, 2020
作者:
W
wangguanzhong
提交者:
GitHub
4月 15, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update doc in export model (#505)
上级
8fd6f220
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
8 addition
and
4 deletion
+8
-4
docs/advanced_tutorials/inference/EXPORT_MODEL.md
docs/advanced_tutorials/inference/EXPORT_MODEL.md
+8
-4
未找到文件。
docs/advanced_tutorials/inference/EXPORT_MODEL.md
浏览文件 @
80b02773
# 模型导出
# 模型导出
训练得到一个满足要求的模型后,如果想要将该模型接入到C++预测库或者Serving服务,需要通过
`tools/export_model.py`
导出该模型。
训练得到一个满足要求的模型后,如果想要将该模型接入到C++预测库或者Serving服务,需要通过
`tools/export_model.py`
导出该模型。
同时,会导出预测时使用的配置文件,路径与模型保存路径相同, 配置文件名为
`infer_cfg.yml`
。
**说明:**
**说明:**
-
导出模型输入为网络输入图像,即原始图片经过预处理后的图像,具体预处理方式可参考配置文件中TestReader部分。各类检测模型的输入格式分别为:
-
**输入部分:**
导出模型输入为网络输入图像,即原始图片经过预处理后的图像,具体预处理方式可参考配置文件中TestReader部分。各类检测模型的输入格式分别为:
| 模型系列名称 | 输入图像预处理方式 | 其他输入信息 |
| 模型系列名称 | 输入图像预处理方式 | 其他输入信息 |
| :---------: | ----------- | ---------- |
| :---------: | ----------- | ---------- |
...
@@ -13,8 +13,12 @@
...
@@ -13,8 +13,12 @@
| RCNN | 归一化,等比例缩放 | 1. im
\_
info: 格式为[input
\_
H, input
\_
W, scale],input为输入图像,scale为
```输入图像大小/原始图像大小```
<br>
2. im
\_
shape:格式为[origin
\_
H, origin
\_
W, 1.], origin为原始图像 |
| RCNN | 归一化,等比例缩放 | 1. im
\_
info: 格式为[input
\_
H, input
\_
W, scale],input为输入图像,scale为
```输入图像大小/原始图像大小```
<br>
2. im
\_
shape:格式为[origin
\_
H, origin
\_
W, 1.], origin为原始图像 |
| RCNN+FPN | 归一化,等比例缩放,对图像填充0使得长宽均为32的倍数 | 1. im
\_
info: 格式为[input
\_
H, input
\_
W, scale],input为输入图像,scale为
```输入图像大小/原始图像大小```
<br>
2. im
\_
shape:格式为[origin
\_
H, origin
\_
W, 1.], origin为原始图像 |
| RCNN+FPN | 归一化,等比例缩放,对图像填充0使得长宽均为32的倍数 | 1. im
\_
info: 格式为[input
\_
H, input
\_
W, scale],input为输入图像,scale为
```输入图像大小/原始图像大小```
<br>
2. im
\_
shape:格式为[origin
\_
H, origin
\_
W, 1.], origin为原始图像 |
| RetinaNet | 归一化,等比例缩放,对图像填充0使得长宽均为128的倍数 | 1. im
\_
info: 格式为[input
\_
H, input
\_
W, scale],input为输入图像,scale为
```输入图像大小/原始图像大小```
<br>
2. im
\_
shape:格式为[origin
\_
H, origin
\_
W, 1.], origin为原始图像 |
| RetinaNet | 归一化,等比例缩放,对图像填充0使得长宽均为128的倍数 | 1. im
\_
info: 格式为[input
\_
H, input
\_
W, scale],input为输入图像,scale为
```输入图像大小/原始图像大小```
<br>
2. im
\_
shape:格式为[origin
\_
H, origin
\_
W, 1.], origin为原始图像 |
| Face | 归一化 | im
\_
shape: 格式为[origin
\_
H, origin
\_
W], origin为原始图像 |
-
导出模型输出统一为NMS的输出,形状为[N, 6], 其中N为预测框的个数,6为[class_id, score, x1, y1, x2, y2].
-
**输出部分:**
导出模型输出统一为NMS的输出,形状为[N, 6], 其中N为预测框的个数,6为[class_id, score, x1, y1, x2, y2]。
-
模型导出不支持模型结构中包含
```fluid.layers.py_func```
的情况。
## 启动参数说明
## 启动参数说明
...
@@ -57,6 +61,6 @@ python tools/export_model.py -c configs/yolov3_darknet.yml \
...
@@ -57,6 +61,6 @@ python tools/export_model.py -c configs/yolov3_darknet.yml \
# 导出SSD模型,输入是3x300x300
# 导出SSD模型,输入是3x300x300
python tools/export_model.py
-c
configs/ssd/ssd_mobilenet_v1_voc.yml
\
python tools/export_model.py
-c
configs/ssd/ssd_mobilenet_v1_voc.yml
\
--output_dir
=
./inference_model
\
--output_dir
=
./inference_model
\
-o
weights
=
https://paddlemodels.bj.bcebos.com/object_detection/ssd_mobilenet_v1_voc.tar
\
-o
weights
=
https://paddlemodels.bj.bcebos.com/object_detection/ssd_mobilenet_v1_voc.tar
\
TestReader.inputs_def.image_shape
=[
3,300,300]
TestReader.inputs_def.image_shape
=[
3,300,300]
```
```
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录