product_recognition.md 1.8 KB
Newer Older
C
cuicheng01 已提交
1 2
# 商品识别

D
dongshuilong 已提交
3
商品识别技术,是现如今应用非常广的一个领域。拍照购物的方式已经被很多人所采纳,无人结算台已经走入各大超市,无人超市更是如火如荼,这背后都是以商品识别技术作为支撑。商品识别技术大概是"商品检测+商品识别"这样的流程,商品检测模块负责检测出潜在的商品区域,商品识别模型负责将商品检测模块检测出的主体进行识别。识别模块多采用检索的方式,根据查询图片和底库图片进行相似度排序获得预测类别。此文档主要对商品图片的特征提取部分进行相关介绍。
C
cuicheng01 已提交
4

D
dongshuilong 已提交
5 6
## 1 算法介绍

D
dongshuilong 已提交
7
算法整体流程,详见[特征学习](./feature_learning.md)整体流程。
D
dongshuilong 已提交
8

D
dongshuilong 已提交
9
整体设置详见: [ResNet50_vd_Aliproduct.yaml](../../../ppcls/configs/Products/ResNet50_vd_Aliproduct.yaml)
D
dongshuilong 已提交
10 11

具体细节如下所示。
C
cuicheng01 已提交
12

D
dongshuilong 已提交
13
### 1.1数据增强
C
cuicheng01 已提交
14

15
- 图像`RandomCrop`到224x224
C
cuicheng01 已提交
16 17 18
- 图像`RandomFlip`
- Normlize:图像归一化  

D
dongshuilong 已提交
19
### 1.2 Backbone的具体设置
C
cuicheng01 已提交
20

D
dongshuilong 已提交
21
具体是用`ResNet50_vd`作为backbone,使用ImageNet预训练模型
C
cuicheng01 已提交
22

D
dongshuilong 已提交
23
### 1.3 Neck部分
C
cuicheng01 已提交
24

D
dongshuilong 已提交
25
加入一个512维的embedding FC层,没有做BatchNorm和激活。
C
cuicheng01 已提交
26

D
dongshuilong 已提交
27
### 1.4 Metric Learning相关Loss的设置
C
cuicheng01 已提交
28

29
目前使用了[CELoss](../../../ppcls/loss/celoss.py)训练, 为了获得更加鲁棒的特征,后续会使用其他Loss参与训练,敬请期待。
C
cuicheng01 已提交
30

D
dongshuilong 已提交
31
## 2 实验结果
C
cuicheng01 已提交
32

D
dongshuilong 已提交
33
<img src="../../images/product/aliproduct.png" style="zoom:50%;" />
C
cuicheng01 已提交
34

35
此方案在Aliproduct[1]数据集上进行实验。此数据集是天池竞赛开源的一个数据集,也是目前开源的最大的商品数据集,其有5万多个标识类别,约250万训练图片。
C
cuicheng01 已提交
36

D
dongshuilong 已提交
37
在此数据上,单模型Top 1 Acc:85.67%。
38 39 40 41

## 3 参考文献

[1] Weakly Supervised Learning with Side Information for Noisy Labeled Images. ECCV, 2020.