Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
dc49a564
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
大约 1 年 前同步成功
通知
115
Star
4999
Fork
1114
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
19
列表
看板
标记
里程碑
合并请求
6
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleClas
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
19
Issue
19
列表
看板
标记
里程碑
合并请求
6
合并请求
6
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
dc49a564
编写于
6月 15, 2021
作者:
B
Bin Lu
提交者:
GitHub
6月 15, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update getting_started_retrieval.md
上级
8f3a5e1f
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
10 addition
and
21 deletion
+10
-21
docs/zh_CN/tutorials/getting_started_retrieval.md
docs/zh_CN/tutorials/getting_started_retrieval.md
+10
-21
未找到文件。
docs/zh_CN/tutorials/getting_started_retrieval.md
浏览文件 @
dc49a564
...
...
@@ -226,26 +226,15 @@ python tools/export_model.py \
通过3.1生成模型结构文件(
`inference.pdmodel`
)和模型权重文件(
`inference.pdiparams`
),通过3.2构建好底库, 然后可以使用预测引擎进行推理:
```
bash
python tools/infer/predict.py
\
--image_file
图片路径
\
--model_file
"./inference/inference.pdmodel"
\
--params_file
"./inference/inference.pdiparams"
\
--use_gpu
=
True
\
--use_tensorrt
=
False
python python/predict_rec.py
\
--Global
.infer_imgs 图片路径
\
--Global
.rec_inference_model_dir
"./inference"
--Global
.use_gpu
=
True
\
--Global
.use_tensorrt
=
False
```
其中:
+
`image_file`
:待预测的图片文件路径,如
`./test.jpeg`
+
`model_file`
:模型结构文件路径,如
`./inference/inference.pdmodel`
+
`params_file`
:模型权重文件路径,如
`./inference/inference.pdiparams`
+
`use_tensorrt`
:是否使用 TesorRT 预测引擎,默认值:
`True`
+
`use_gpu`
:是否使用 GPU 预测,默认值:
`True`
+
`enable_mkldnn`
:是否启用
`MKL-DNN`
加速,默认为
`False`
。注意
`enable_mkldnn`
与
`use_gpu`
同时为
`True`
时,将忽略
`enable_mkldnn`
,而使用GPU运行。
+
`resize_short`
: 对输入图像进行等比例缩放,表示最短边的尺寸,默认值:
`256`
+
`resize`
: 对
`resize_short`
操作后的进行居中裁剪,表示裁剪的尺寸,默认值:
`224`
+
`enable_calc_topk`
: 是否计算预测结果的Topk精度指标,默认为
`False`
,
+
`gt_label_path`
: 图像文件名以及真值标签文件,当
`enable_calc_topk`
为True时生效,用于读取待预测的图像列表及其标签。
**注意**
: 如果使用
`Transformer`
系列模型,如
`DeiT_***_384`
,
`ViT_***_384`
等,请注意模型的输入数据尺寸,需要设置参数
`resize_short=384`
,
`resize=384`
。
*
如果你希望评测模型速度,建议使用该脚本(
`tools/infer/predict.py`
),同时开启TensorRT加速预测。
+
`Global.infer_imgs`
:待预测的图片文件路径,如
`./test.jpeg`
+
`Global.rec_inference_model_dir`
:模型结构文件路径,如
`./inference/`
+
`Global.use_tensorrt`
:是否使用 TesorRT 预测引擎,默认值:
`True`
+
`Global.use_gpu`
:是否使用 GPU 预测,默认值:
`True`
+
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录