未验证 提交 adb0a78c 编写于 作者: B Bin Lu 提交者: GitHub

Update feature_extraction.md

上级 bf92ff47
...@@ -78,7 +78,7 @@ Loss: ...@@ -78,7 +78,7 @@ Loss:
# 4.训练、评估、推理 # 4.训练、评估、推理
下面以`ppcls/configs/Products/ResNet50_vd_SOP.yaml`为例,介绍模型的训练、评估、推理过程 下面以`ppcls/configs/Products/ResNet50_vd_SOP.yaml`为例,介绍模型的训练、评估、推理过程
## 4.1 数据准备 ## 4.1 数据准备
首先,下载SOP数据集, 数据链接 首先,下载SOP数据集, 数据链接: https://cvgl.stanford.edu/projects/lifted_struct/
## 4.2 训练 ## 4.2 训练
- 单机单卡训练 - 单机单卡训练
...@@ -96,7 +96,7 @@ python -m paddle.distributed.launch ...@@ -96,7 +96,7 @@ python -m paddle.distributed.launch
## 4.3 评估 ## 4.3 评估
- 单卡评估 - 单卡评估
``` ```
python tools/eval.py -c ppcls/configs/ResNet50_vd_SOP.yaml -o Global.pretrained_model = "output/ReModel/best_model" python tools/eval.py -c ppcls/configs/ResNet50_vd_SOP.yaml -o Global.pretrained_model="output/ReModel/best_model"
``` ```
- 多卡评估 - 多卡评估
``` ```
...@@ -109,9 +109,9 @@ python -m paddle.distributed.launch ...@@ -109,9 +109,9 @@ python -m paddle.distributed.launch
推理过程包括两个步骤: 1) 导出推理模型; 2) 获取特征向量 推理过程包括两个步骤: 1) 导出推理模型; 2) 获取特征向量
### 4.4.1 导出推理模型 ### 4.4.1 导出推理模型
``` ```
python tools/export_model -c ppcls/configs/ResNet50_vd_SOP.yaml -o Global.pretrained_model = "output/ReModel/best_model" python tools/export_model -c ppcls/configs/ResNet50_vd_SOP.yaml -o Global.pretrained_model="output/ReModel/best_model"
``` ```
生成的推理模型位于inference目录,名字为inference.pd* 生成的推理模型位于`inference`目录,名字为`inference.pd*`
### 4.4.2 获取特征向量 ### 4.4.2 获取特征向量
``` ```
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册