[发明专利]图像识别的方法和装置

说明书

本公开涉及一种图像识别的方法和装置，该方法包括：获取待识别物品的物品图像；沿物品图像的图像中心按照多个旋转角度旋转物品图像得到的多个旋转图像；通过预设分类模型从多个旋转图像中，确定出第一图像标记以及第一图像标记对应的置信度，其中，第一图像标记用于标示多个旋转图像中属于目标类别的待识别物品的图像；置信度表示第一图像标记标示出的待识别物品属于目标类别的可信度；根据置信度和第一图像标记在物品图像中确定目标图像标记，其中，目标图像标记用于标记物品图像中确定属于目标类别的待识别物品的图像。

技术领域

本公开涉及图像处理领域，具体地，涉及一种图像识别的方法和装置。

背景技术

随着经济的发展，往返于贸易港口、火车站、航空站等海关监管区域的旅客、商品越来越多，而在旅客行李中隐藏的危险物品和走私物品不仅严重威胁着旅客的生命财产安全，更危害了国家安全和利益。

目前国内的安检系统主要依靠安检工作人员通过人眼进行识别，其中，旅客的行李或者包裹在通过传送带或安检门时，透过X射线扫描出的图像会出现在与设备连接的显示屏幕上，安检人员通过观察屏幕上的图像、色块等对其中的危险物品或者走私物品进行识别。

但是，由于现有方法是由人工操作的，因此，在对危险物品或者走私物品进行识别难免存在漏检或者错检的问题，从而无法准确的识别危险物品或者走私物品。

发明内容

本公开的目的是提供一种图像识别的方法和装置，能够准确识别待识别物品的图像。

为了实现上述目的，本公开提供一种图像识别的方法，包括：获取待识别物品的物品图像；沿所述物品图像的图像中心按照多个旋转角度旋转所述物品图像得到的多个旋转图像；通过预设分类模型从所述多个旋转图像中，确定出第一图像标记以及所述第一图像标记对应的置信度，其中，所述第一图像标记用于标示所述多个旋转图像中属于目标类别的待识别物品的图像；所述置信度表示所述第一图像标记标示出的待识别物品属于所述目标类别的可信度；根据所述置信度和所述第一图像标记在所述物品图像中确定目标图像标记，其中，所述目标图像标记用于标记所述物品图像中确定属于所述目标类别的待识别物品的图像。

可选地，所述根据所述置信度和所述第一图像标记在所述物品图像中标记出目标物品图像包括：获取所述第一图像标记所在旋转图像对应的旋转方向；确定与所述旋转方向相反的反转方向；沿所述第一图像标记所在旋转图像的图像中心由所述反转方向按照所述旋转角度旋转所述第一图像标记，并将旋转后的第一图像标记叠加到所述物品图像上；根据旋转后的第一图像标记和所述置信度在所述物品图像中标记出目标物品图像。

可选地，所述第一图像标记包括图像标记框，所述根据旋转后的第一图像标记和所述置信度在所述物品图像中标记出目标物品图像包括：确定所述图像标记框中是否包括存在交集的图像标记框；在所述图像标记框中包括存在交集的图像标记框时，从存在交集的图像标记框中确定置信度最大的第一标记框；根据所述第一标记框与第二标记框的交集确定匹配值；所述第二标记框为存在交集的图像标记框中除所述第一标记框之外的其他标记框；在所述匹配值大于或者等于预设匹配值时，确定所述第一标记框为所述目标标记框且所述第二标记框不是所述目标标记框；在所述匹配值小于所述预设匹配值时，确定所述第一标记框和所述第二标记框都为所述目标标记框。

可选地，所述根据所述第二标记框与所述第一标记框的交集确定匹配值包括：获取所述第二标记框与所述第一标记框的交集的第一面积；获取所述第二标记框与所述第一标记框的并集的第二面积；根据所述第一面积和所述第二面积通过以下公式确定所述匹配值；

其中，P为所述匹配值，S1为所述第一面积，S2为所述第二面积。