[发明专利]一种融合注意力图的商品分类方法

摘要

本发明公开了一种融合注意力图的商品分类方法。收集一张清晰、完整的商品正面照片作为模板图像，进行扩充得到训练图像，训练图像与模板图像进行匹配获得注意力图，将RGB三通道与注意力图叠加形成四通道，组成训练数据，训练获得图像分类模型，将待分类商品图像的四通道数据输入图像分类模型，得到待分类商品图像的预测分类与对应分数。本发明方法利用能够放大商品图案细微特征对分类模型影响力、抑制图案非显著性特征的注意力图参与训练，提升了商品图像分类的正确率。

主权项

1.一种融合注意力图的商品分类方法，其特征在于包含以下步骤：1)对每一类别的商品，收集一张清晰、完整的正面照片作为模板图像；2)对步骤1)中每类商品的单张模板图像进行扩充，得到每类商品的训练图像；3)对步骤2)中的每张训练图像与生成训练图像的模板图像进行匹配，根据匹配关系绘制注意力图；4)将步骤2)中训练图像的RGB三通道与步骤3)中生成的注意力图叠加形成四通道，四通道和训练图像自身带有分类标签组成训练数据，训练获得图像分类模型；5)针对待分类商品图像按照步骤2)和3)相同方式处理获得注意力图，将注意力图与待分类商品图像自身的RGB三通道叠加成四通道，将四通道数据输入步骤4)训练得到的图像分类模型，得到待分类商品图像的预测分类与对应分数。