Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
7fc118c3
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看板
未验证
提交
7fc118c3
编写于
6月 16, 2021
作者:
B
Bin Lu
提交者:
GitHub
6月 16, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update getting_started_retrieval.md
上级
4864a465
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
7 addition
and
15 deletion
+7
-15
docs/zh_CN/tutorials/getting_started_retrieval.md
docs/zh_CN/tutorials/getting_started_retrieval.md
+7
-15
未找到文件。
docs/zh_CN/tutorials/getting_started_retrieval.md
浏览文件 @
7fc118c3
...
...
@@ -87,7 +87,7 @@ export CUDA_VISIBLE_DEVICES=0,1,2,3
python
-m
paddle.distributed.launch
\
--gpus
=
"0,1,2,3"
\
tools/train.py
\
-c
.
/configs/quick_start/ResNet50_vd_finetune_retrieval.yaml
-c
ppcls
/configs/quick_start/ResNet50_vd_finetune_retrieval.yaml
```
### 2.2 模型微调
...
...
@@ -100,7 +100,7 @@ export CUDA_VISIBLE_DEVICES=0,1,2,3
python -m paddle.distributed.launch \
--gpus="0,1,2,3" \
tools/train.py \
-c
.
/configs/quick_start/ResNet50_vd_finetune_retrieval.yaml \
-c
ppcls
/configs/quick_start/ResNet50_vd_finetune_retrieval.yaml \
-o Arch.Backbone.pretrained=True
```
...
...
@@ -114,13 +114,10 @@ export CUDA_VISIBLE_DEVICES=0,1,2,3
python -m paddle.distributed.launch \
--gpus="0,1,2,3" \
tools/train.py \
-c
.
/configs/quick_start/ResNet50_vd_finetune_retrieval.yaml \
-o Global.checkpoints="./output/RecModel/
ppcls_
epoch_5" \
-c
ppcls
/configs/quick_start/ResNet50_vd_finetune_retrieval.yaml \
-o Global.checkpoints="./output/RecModel/epoch_5" \
```
其中配置文件不需要做任何修改,只需要在训练时设置
`checkpoints`
参数与
`last_epoch`
参数即可,该参数表示加载的断点权重文件路径,使用该参数会同时加载保存的模型参数权重和学习率、优化器等信息,详见
[
1.3 模型恢复训练
](
#1.3
)
。
### 2.4 模型评估
可以通过以下命令进行模型评估。
...
...
@@ -129,12 +126,10 @@ python -m paddle.distributed.launch \
python.
-m
paddle.distributed.launch
\
--gpus
=
"0,1,2,3"
\
tools/eval.py
\
-c
.
/configs/quick_start/ResNet50_vd_finetune_retrieval.yaml
\
-c
ppcls
/configs/quick_start/ResNet50_vd_finetune_retrieval.yaml
\
-o
Global.pretrained_model
=
"./output/RecModel/best_model"
\
```
参数说明详见
[
1.4 模型评估
](
#1.4
)
。
<a
name=
"model_inference"
></a>
## 3. 使用inference模型进行模型推理
### 3.1 导出推理模型
...
...
@@ -148,11 +143,8 @@ python tools/export_model.py \
--Global
.save_inference_dir ./inference
\
```
其中,
`--pretrained_model`
用于指定模型文件路径,该路径仍无需包含模型文件后缀名(如
[
1.3 模型恢复训练
](
#1.3
)
),
`--save_inference_dir`
用于指定转换后模型的存储路径。
**注意**
:
1.
`--save_inference_dir`
表示输出的inference模型文件夹路径,若
`--save_inference_dir=./inference`
,则会在
`inference`
文件夹下生成
`inference.pdiparams`
、
`inference.pdmodel`
和
`inference.pdiparams.info`
文件。
2.
可以通过设置参数
`--img_size`
指定模型输入图像的
`shape`
,默认为
`224`
,表示图像尺寸为
`224*224`
,请根据实际情况修改。
其中,
`--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 构建底库
通过检索方式来进行图像识别,需要构建底库。底库构建方式如下:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录