[发明专利]一种小样本商品图像分类方法、装置、设备及存储介质

摘要

本发明涉及一种小样本商品图像分类方法、装置、设备及存储介质。该方法包括如下步骤：获取目标商品图像；将所述目标商品图像与支持集图像输入至训练好的商品分类模型，得到该商品图像与所述支持集图像中每个图像之间的相似度，其中，所述商品分类模型包括嵌入模块和度量模块，所述嵌入模块包括相互连接的第一卷积层和第一ECA层，所述度量模块包括相互连接的第二卷积层和全连接层；根据最大的所述相似度，得到该目标商品的类别。该方法的嵌入模块融入了ECA模块，能够提取更加精确的特征图，从而使得后面度量模块分类更加有效。同时使得融入了ECA模块的度量模块能够在比较特征相似度的同时利用不同类别图像特征之间的关系，从而提高分类的准确率。