Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
c4e1e3c7
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看板
未验证
提交
c4e1e3c7
编写于
6月 16, 2021
作者:
B
Bin Lu
提交者:
GitHub
6月 16, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update getting_started_retrieval.md
上级
86348dc4
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
16 addition
and
10 deletion
+16
-10
docs/zh_CN/tutorials/getting_started_retrieval.md
docs/zh_CN/tutorials/getting_started_retrieval.md
+16
-10
未找到文件。
docs/zh_CN/tutorials/getting_started_retrieval.md
浏览文件 @
c4e1e3c7
...
...
@@ -139,23 +139,29 @@ python. -m paddle.distributed.launch \
```
bash
python tools/export_model.py
\
--Global
.pretrained_model ./output/RecModel/best_model
\
--Global
.save_inference_dir ./inference
\
-c
ppcls/configs/quick_start/ResNet50_vd_finetune_retrieval.yaml
\
-o
Global.pretrained_model
=
./output/RecModel/best_model
\
-o
Global.save_inference_dir
=
./inference
\
```
其中,
`--Global.pretrained_model`
用于指定模型文件路径,该路径仍无需包含模型文件后缀名(如
[
1.3 模型恢复训练
](
#1.3
)
),
`--Global.save_inference_dir`
用于指定转换后模型的存储路径。
若
`--save_inference_dir=./inference`
,则会在
`inference`
文件夹下生成
`inference.pdiparams`
、
`inference.pdmodel`
和
`inference.pdiparams.info`
文件。
### 3.2 构建底库
通过检索方式来进行图像识别,需要构建底库。底库构建方式如下:
通过检索方式来进行图像识别,需要构建底库。
首先, 将生成的模型拷贝到deploy目录下,并进入deploy目录:
```
bash
mv
./inference ./deploy
cd
deploy
```
其次,构建底库,命令如下:
```
bash
python python/build_gallery.py
-c
configs/build_flowers.yaml
\
-o
Global.rec_inference_model_dir
".
.
/inference"
\
-o
IndexProcess.index_path
"../dataset
/index"
\
-o
IndexProcess.image_root:
"../dataset"
\
-o
IndexProcess.data_file:
"../dataset/train_list.txt"
-o
Global.rec_inference_model_dir
"./inference"
\
-o
IndexProcess.index_path
"../dataset/flowers102
/index"
\
-o
IndexProcess.image_root:
"../dataset
/flowers102/
"
\
-o
IndexProcess.data_file:
"../dataset/
flowers102/
train_list.txt"
```
其中
+
`Global.rec_inference_model_dir`
:3.1生成的推理模型的路径
...
...
@@ -171,12 +177,12 @@ python python/build_gallery.py
python python/predict_rec.py
\
-c
configs/inference_flowers.yaml
\
-o
Global.infer_imgs 图片路径
\
-o
Global.rec_inference_model_dir
"./inference"
-o
Global.rec_inference_model_dir
".
.
/inference"
-o
Global.use_gpu
=
True
\
-o
Global.use_tensorrt
=
False
```
其中:
+
`Global.infer_imgs`
:待预测的图片文件路径,如
`./test.jpeg`
+
`Global.rec_inference_model_dir`
:模型结构文件路径,如
`./inference/`
+
`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.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录