PaddleClas/docs/zh_CN/application/feature_learning.md

29 lines
2.5 KiB
Markdown
Raw Normal View History

# 特征学习
2021-06-17 20:40:50 +08:00
此部分主要是针对特征学习的训练模式进行说明,即`RecModel`的训练模式。主要是为了支持车辆识别车辆细分类、ReID、Logo识别、动漫人物识别、商品识别等特征学习的应用。与在`ImageNet`上训练普通的分类网络不同的是,此特征学习部分,主要有以下特征
- 支持对`backbone`的输出进行截断,即支持提取任意中间层的特征信息
- 支持在`backbone`的feature输出层后添加可配置的网络层即`Neck`部分
2021-06-17 20:40:50 +08:00
- 支持`ArcFace Loss`等`metric learning` 相关loss函数提升特征学习能力
2021-06-18 15:01:45 +08:00
## 1 整体流程
2021-06-17 20:40:50 +08:00
![](../../images/recognition/rec_pipeline.png)
2021-06-17 20:40:50 +08:00
特征学习的整体结构如上图所示主要包括数据增强、Backbone的设置、Neck、Metric Learning等几大部分。其中`Neck`部分为自由添加的网络层如添加的embedding层等当然也可以不用此模块。训练时利用`Metric Learning`部分的Loss对模型进行优化。预测时一般来说默认以`Neck`部分的输出作为特征输出。
2021-06-17 20:40:50 +08:00
针对不同的应用可以根据需要对每一部分自由选择。每一部分的具体配置如数据增强、Backbone、Neck、Metric Learning相关Loss等设置详见具体应用[车辆识别](./vehicle_recognition.md)、[Logo识别](./logo_recognition.md)、[动漫人物识别](./cartoon_character_recognition.md)、[商品识别](./product_recognition.md)
2021-06-18 15:01:45 +08:00
## 2 配置文件说明
2021-06-17 20:40:50 +08:00
配置文件说明详见[yaml配置文件说明文档](../tutorials/config.md)。其中模型结构配置,详见文档中**识别模型结构配置**部分。
2021-06-29 15:45:39 +08:00
## 3 预训练模型
- [车辆细分类](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/pretrain/vehicle_cls_ResNet50_CompCars_v1.1_pretrained.pdparams)
- [车辆ReID](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/pretrain/vehicle_reid_ResNet50_VERIWild_v1.0_pretrained.pdparams)
- [动漫人物](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/pretrain/cartoon_rec_ResNet50_iCartoon_v1.0_pretrained.pdparams)
- [Logo识别](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/pretrain/logo_rec_ResNet50_Logo3K_v1.0_pretrained.pdparams)
2021-06-29 15:58:53 +08:00
- 商品识别: [Inshop](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/pretrain/product_ResNet50_vd_Inshop_pretrained_v1.0.pdparams)、[Aliproduct](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/pretrain/product_ResNet50_vd_Aliproduct_v1.0_pretrained.pdparams)
2021-06-29 15:45:39 +08:00