[发明专利]零售场景下的商品识别检测计数方法及系统

权利要求书

1.一种零售场景下的商品识别检测计数方法，其特征在于，包括：

获取待检测的零售场景图像；

利用预先训练好的检测模型处理所述待检测的零售场景图像，获取图像中包含的商品的类型及数量；其中，所述预先训练好的检测模型由训练集训练得到，所述训练集包括多张包含有零售商品的照片以及在所述照片中标注商品类别信息和商品数量信息的标签；其中，所述训练集中将零售场景照片的标注转化为COCO数据集采用的格式。

2.根据权利要求1所述的零售场景下的商品识别检测计数方法，其特征在于，使用残差网络与特征金字塔结构作为所述检测模型的主干网络；所述主杆网络中，对图像进行卷积操作后利用残差块获得不同尺度的特征图，利用特征金字塔结构对不同尺度的特征图进行特征融合，得到融合特征图。

3.根据权利要求2所述的零售场景下的商品识别检测计数方法，其特征在于，所述主干网络连接候选框生成网络，融合特征图输入到候选框生成网络对进行候选框的生成。

4.根据权利要求3所述的零售场景下的商品识别检测计数方法，其特征在于，所述候选框生成网络包括两条支路，一条是对生成的候选框进行分类，对候选框内所包含图像进行判断，将其分为前景背景；另一条支路是对候选框进行初步调整，得到初步的目标建议；所述候选框生成网络最终整合两条支路的检测建议，通过计算得到最终的检测目标建议。

5.根据权利要求4所述的零售场景下的商品识别检测计数方法，其特征在于，所述候选框生成网络连接级联的目标检测网络结构，依据检测目标建议，进行多阶段的定位计数与分类，生成最终的目标检测框，其中包含最终的商品类别和所包含商品的数量。

6.根据权利要求1-5任一项所述的零售场景下的商品识别检测计数方法，其特征在于，在训练所述检测模型时，使用多任务损失函数来实现端到端的训练，多任务损失函数由分类损失、回归损失、计数损失构成。

7.一种零售场景下的商品识别检测计数系统，其特征在于，包括：

获取模块，用于获取待检测的零售场景图像；

检测模块，用于利用预先训练好的检测模型处理所述待检测的零售场景图像，获取图像中包含的商品的类型及数量；其中，所述预先训练好的检测模型由训练集训练得到，所述训练集包括多张包含有零售商品的照片以及在所述照片中标注商品类别信息和商品数量信息的标签；其中，所述训练集中将零售场景照片的标注转化为COCO数据集采用的格式。

8.一种非暂态计算机可读存储介质，其特征在于，所述非暂态计算机可读存储介质用于存储计算机指令，所述计算机指令被处理器执行时，实现如权利要求1-6任一项所述的零售场景下的商品识别检测计数方法。

9.一种计算机程序产品，其特征在于，包括计算机程序，所述计算机程序当在一个或多个处理器上运行时，用于实现如权利要求1-6任一项所述的零售场景下的商品识别检测计数方法。

10.一种电子设备，其特征在于，包括：处理器、存储器以及计算机程序；其中，处理器与存储器连接，计算机程序被存储在存储器中，当电子设备运行时，所述处理器执行所述存储器存储的计算机程序，以使电子设备执行实现如权利要求1-6任一项所述的零售场景下的商品识别检测计数方法的指令。