Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
5aca5d0f
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看板
未验证
提交
5aca5d0f
编写于
10月 29, 2021
作者:
B
Bin Lu
提交者:
GitHub
10月 29, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update feature_extraction.md
上级
c33f97ee
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
7 addition
and
4 deletion
+7
-4
docs/zh_CN_tmp/image_recognition_pipeline/feature_extraction.md
...h_CN_tmp/image_recognition_pipeline/feature_extraction.md
+7
-4
未找到文件。
docs/zh_CN_tmp/image_recognition_pipeline/feature_extraction.md
浏览文件 @
5aca5d0f
...
...
@@ -43,9 +43,9 @@ Arch:
-
**infer_output_key**
: 推理时需要用到的Tensor的key, 训练模型下,网络会以字典的形式输出features和logits. 识别任务中,推理时只需要用到features即可
-
**infer_output_key**
: 推理时是否需要加softmax。为了和分类任务的统一,分类任务推理时需要加softmax操作,识别任务不需要
-
**Backbone**
: 骨干网络, 此处选用的是经过SSLD蒸馏之后的预训练模型
-
**BackboneStopLayer**
:
该处用以指示在哪儿截断
-
**Neck**
: 输出512维的特征向量
-
**Head**
: 采用ArcMargin, 此处可以依据训练数据修改类别数class_num,
以及超参数margin和scale
-
**BackboneStopLayer**
:
该处用以指示在何处截断网络,关于TheseLayer的用法,请参考文档xxx
-
**Neck**
:
用以进行特征维度转换,此处
输出512维的特征向量
-
**Head**
: 采用ArcMargin, 此处可以依据训练数据修改类别数class_num, 以及超参数margin和scale
## Loss构成
### 单Loss示例
...
...
@@ -58,7 +58,7 @@ Loss:
- CELoss:
weight: 1.0
```
可以看到此处选用的是CELoss, 结合Head部分的ArcMargin, 因此使用的是
ArcFace
中的算法
可以看到此处选用的是CELoss, 结合Head部分的ArcMargin, 因此使用的是
[
ArcFace
](
https://arxiv.org/abs/1801.07698
)
中的算法
### 组合Loss示例
```
...
...
@@ -79,6 +79,7 @@ Loss:
下面以
`ppcls/configs/Products/ResNet50_vd_SOP.yaml`
为例,介绍模型的训练、评估、推理过程
## 4.1 数据准备
首先,下载SOP数据集, 数据链接
## 4.2 训练
-
单机单卡训练
```
...
...
@@ -121,3 +122,5 @@ python tools/export_model -c xxx -o Global.pretrained_model = xxxx
cd deploy
python python/inference_rec.py -c configs/ O rec_inference_model_dir: "../inference/inference"
```
输出
```
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录