[发明专利]识别待检测货物类别的方法和装置

权利要求书

1.一种识别待检测货物类别的方法，其特征在于，包括：

采集整托待检测货物的正面图像，将所述正面图像输入分割模型，得到分割后的待检测货箱图像；

其中，所述分割模型是基于整托货物的样本正面图像以及预先在所述样本正面图像上标注的各货箱的四角坐标进行训练后得到的；

选取所述分割后的待检测货箱图像中的任意一幅，输入识别模型，得到所述待检测货箱图像对应的第一货物类别；

其中，所述识别模型是基于样本货箱图像以及预先确定的货物类别标签进行训练后得到的。

2.根据权利要求1所述的识别待检测货物类别的方法，其特征在于，还包括：

根据预先存储的货物类别与类别标识的对应关系，获取所述第一货物类别对应的类别标识。

3.根据权利要求1所述的识别待检测货物类别的方法，其特征在于，还包括：

采用激光测距仪采集整托货物的高度，基于所述高度和所述分割后的待检测货箱图像得到整托货物的层数，计算单个待检测货箱的高度；

根据预先存储的货物类别与货箱的高度对应关系，获取所述第一货物类别对应的货箱的高度；

若所述单个待检测货箱的高度与所述第一货物类别对应的货箱的高度相等，则维持待检测货物的类别为第一货物类别；

若所述单个待检测货箱的高度与所述第一货物类别对应的货箱的高度不同，则重新进行货物类别识别。

4.根据权利要求1-3中任一项所述的识别待检测货物类别的方法，其特征在于，所述选取所述分割后的待检测货箱图像中的任意一幅，输入识别模型，得到所述待检测货箱图像对应的第一货物类别，具体包括：

选取所述分割后的待检测货箱图像中的任意一幅，输入识别模型，输出输入图像对应于各货物类别的概率值；

选择概率值最大的货物类别作为待检测货箱图像对应的第一货物类别。

5.一种识别待检测货物类别的装置，其特征在于，包括：

第一输入单元，用于采集整托待检测货物的正面图像，将所述正面图像输入分割模型，得到分割后的待检测货箱图像；其中，所述分割模型是基于整托货物的样本正面图像以及预先在所述样本正面图像上标注的各货箱的四角坐标进行训练后得到的；

第二输入单元，用于选取所述分割后的待检测货箱图像中的任意一幅，输入识别模型，得到所述待检测货箱图像对应的第一货物类别；其中，所述识别模型是基于样本货箱图像以及预先确定的货物类别标签进行训练后得到的。

6.根据权利要求5所述的识别待检测货物类别的装置，其特征在于，还包括：标识单元，

用于根据预先存储的货物类别与类别标识的对应关系，获取所述第一货物类别对应的类别标识。

7.根据权利要求5所述的识别待检测货物类别的装置，其特征在于，还包括：校验单元，

用于采用激光测距仪采集整托货物的高度，基于所述高度和所述分割后的待检测货箱图像得到整托货物的层数，计算单个待检测货箱的高度；

根据预先存储的货物类别与货箱的高度对应关系，获取所述第一货物类别对应的货箱的高度；

若所述单个待检测货箱的高度与所述第一货物类别对应的货箱的高度相等，则维持待检测货物的类别为第一货物类别；

若所述单个待检测货箱的高度与所述第一货物类别对应的货箱的高度不同，则重新进行货物类别识别。

8.根据权利要求5-7中任一项所述的识别待检测货物类别的装置，其特征在于，所述选取所述分割后的待检测货箱图像中的任意一幅，输入识别模型，得到所述待检测货箱图像对应的第一货物类别，具体包括：

选取所述分割后的待检测货箱图像中的任意一幅，输入识别模型，输出输入图像对应于各货物类别的概率值；

选择概率值最大的货物类别作为待检测货箱图像对应的第一货物类别。

9.一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现如权利要求1-4中任一项所述的识别待检测货物类别的方法的步骤。

10.一种非暂态计算机可读存储介质，其上存储有计算机程序，其特征在于，该计算机程序被处理器执行时实现如权利要求1-4中任一项所述的识别待检测货物类别的方法的步骤。