Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
6d9d8070
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看板
未验证
提交
6d9d8070
编写于
11月 08, 2021
作者:
B
Bin Lu
提交者:
GitHub
11月 08, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update feature_extraction.md
上级
5fc825b0
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
10 addition
and
10 deletion
+10
-10
docs/zh_CN/image_recognition_pipeline/feature_extraction.md
docs/zh_CN/image_recognition_pipeline/feature_extraction.md
+10
-10
未找到文件。
docs/zh_CN/image_recognition_pipeline/feature_extraction.md
浏览文件 @
6d9d8070
...
...
@@ -15,14 +15,14 @@
在PP-Shitu中, 我们采用
[
PP_LCNet_x2_5
](
../models/PP-LCNet.md
)
作为骨干网络, Neck部分选用Linear Layer, Head部分选用
[
ArcMargin
](
https://arxiv.org/abs/1801.07698
)
, Loss部分选用CELoss,详细的配置文件见
[
通用商品识别配置文件
](
../../../ppcls/configs/GeneralRecognition/GeneralRecognition_PPLCNet_x2_5.yaml
)
。其中,训练数据为如下7个公开数据集的汇总:
| 数据集 | 数据量 | 类别数 | 场景 | 数据集地址 |
| :------------: | :-------------: | :-------: | :-------: | :--------: |
| Aliproduct | 2498771 | 50030 | 商品 |
[
地址
](
https://
www.objects365.org/overview.html
)
|
| GLDv2 | 1580470 | 81313 | 地标 |
[
地址
](
https://
cocodataset.org/
)
|
| VeRI-Wild | 277797 | 30671 | 车辆 |
[
地址
](
https://github.com/
luxiangju-PersonAI/iCartoonFace
)
|
| Aliproduct | 2498771 | 50030 | 商品 |
[
地址
](
https://
retailvisionworkshop.github.io/recognition_challenge_2020/
)
|
| GLDv2 | 1580470 | 81313 | 地标 |
[
地址
](
https://
github.com/cvdfoundation/google-landmark
)
|
| VeRI-Wild | 277797 | 30671 | 车辆 |
[
地址
](
https://github.com/
guishijin/veri-wild
)
|
| LogoDet-3K | 155427 | 3000 | Logo |
[
地址
](
https://github.com/Wangjing1551/LogoDet-3K-Dataset
)
|
| iCartoonFace | 389678 | 5013 | 动漫人物 |
[
地址
](
http
s://rpc-dataset.github.io/
)
|
| SOP | 59551 | 11318 | 商品 |
[
地址
](
https://
rpc-dataset.github.io
/
)
|
| Inshop | 25882 | 3997 | 商品 |
[
地址
](
http
s://rpc-dataset.github.io/
)
|
|
**Total**
|
**5M**
|
**185K**
| ---- |
[
地址
](
https://rpc-dataset.github.io/
)
|
| iCartoonFace | 389678 | 5013 | 动漫人物 |
[
地址
](
http
://challenge.ai.iqiyi.com/detail?raceId=5def69ace9fcf68aef76a75d
)
|
| SOP | 59551 | 11318 | 商品 |
[
地址
](
https://
cvgl.stanford.edu/projects/lifted_struct
/
)
|
| Inshop | 25882 | 3997 | 商品 |
[
地址
](
http
://mmlab.ie.cuhk.edu.hk/projects/DeepFashion.html
)
|
|
**Total**
|
**5M**
|
**185K**
| ---- |
[
地址
](
----
)
|
最终的模型效果如下表所示:
| 模型 | Aliproduct | VeRI-Wild | LogoDet-3K | iCartoonFace | SOP | Inshop | Latency(ms) |
...
...
@@ -34,7 +34,7 @@ PP-LCNet-2.5x | 0.839 | 0.888 | 0.861 | 0.841 | 0.793 | 0.892 | 5.0
# 4. 自定义特征提取
自定义特征提取,是指依据自己的任务,重新训练特征提取模型。主要包含如下四个步骤:1)数据准备;2)模型训练;3)模型评估;4)模型推理。
## 4.1 数据准备
首先,需要基于任务的需要,定制自己的数据集。数据
的制作方法,参见
首先,需要基于任务的需要,定制自己的数据集。数据
集格式参见
[
格式说明
](
https://github.com/PaddlePaddle/PaddleClas/blob/develop/docs/zh_CN/data_preparation/recognition_dataset.md#%E6%95%B0%E6%8D%AE%E9%9B%86%E6%A0%BC%E5%BC%8F%E8%AF%B4%E6%98%8E
)
## 4.2 模型训练
-
单机单卡训练
...
...
@@ -49,7 +49,7 @@ python -m paddle.distributed.launch
--gpus
=
"0,1,2,3"
tools/train.py
-c
ppcls/configs/GeneralRecognition/GeneralRecognition_PPLCNet_x2_5.yaml
```
注意:
配置文件中默认采用
`在线评估`
的方式,如果你想加快训练速度,去除
`在线评估`
,只需要在上述命令后面,增加
`-o eval_during_train=False`
。
**注意:**
配置文件中默认采用
`在线评估`
的方式,如果你想加快训练速度,去除
`在线评估`
,只需要在上述命令后面,增加
`-o eval_during_train=False`
。
训练完毕后,在output目录下会生成最终模型文件
`latest.pd*`
,
`best_model.pd*`
和训练日志文件
`train.log`
。
## 4.3 模型评估
...
...
@@ -68,7 +68,7 @@ python -m paddle.distributed.launch
-c
ppcls/configs/GeneralRecognition/GeneralRecognition_PPLCNet_x2_5.yaml
-o
Global.pretrained_model
=
"output/RecModel/best_model"
```
**推荐:**
使用多卡评估,可以利用多卡快速计算
得到整体数据集的特征集合,能够加速评估的过程。
**推荐:**
建议使用多卡评估。多卡评估方式可以利用多卡并行计算快速
得到整体数据集的特征集合,能够加速评估的过程。
## 4.4 模型推理
推理过程包括两个步骤: 1)导出推理模型, 2)获取特征向量
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录