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

权利要求书

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

采集整托待检测货物的正面图像和顶部图像；

将所述正面图像和所述顶部图像进行白平衡处理，得到白平衡正面图像和白平衡顶部图像；

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

将所述白平衡顶部图像输入第二分割模型，得到分割后的顶部货箱图像和对应所述顶部货箱图像的空缺层数；其中，所述第二分割模型是基于整托货物的样本白平衡顶部图像以及预先在所述样本白平衡顶部图像上标注的各货箱的四角坐标标签和在所述样本白平衡顶部图像上标注的各货箱的空缺层数标签进行训练得到的；

基于所述分割后的正面货箱图像、所述分割后的顶部货箱图像和对应所述顶部货箱图像的空缺层数确定整托待识别货物的数量。

2.根据权利要求1所述的识别待检测货物数量的方法，其特征在于，所述将所述正面图像和所述顶部图像进行白平衡处理，得到白平衡正面图像和白平衡顶部图像，具体包括：

基于所述正面图像和所述顶部图像确定待处理图像；

将所述待处理图像的原始RGB直方图进行量化，形成量化后的RGB直方图；

确定所述量化后的RGB直方图中各色阶的参考白和参考黑；

基于所述各色阶的参考白和参考黑对量化后的RGB直方图进行重新量化，得到最终RGB直方图；

确定所述最终RGB直方图中各色阶的平均值；

基于任一色阶的平均值确定所述任一色阶的偏色值；

对所述原始RGB直方图用每一色阶的偏色值进行校正，得到白平衡后的待处理图像；

基于所述白平衡后的待处理图像确定白平衡正面图像和白平衡顶部图像。

3.根据权利要求2所述的识别待检测货物数量的方法，其特征在于，所述确定所述量化后的RGB直方图中各色阶的参考白和参考黑，具体包括：

取所述量化后的RGB直方图中任一色阶的最大值和次大值进行平均，得到待处理图像的所述任一色阶的参考白；

取所述量化后的RGB直方图中任一色阶的最小值和次小值进行平均，得到待处理图像的所述任一色阶的参考黑；

所述基于所述任一色阶的平均值确定所述任一色阶的偏色值，具体包括:

确定与所有色阶的平均值形成的RGB值亮度相同的中性灰；

计算任一色阶的平均值与所述中性灰之间的差值，作为所述任一色阶的偏色值。

4.根据权利要求1-3中任一项所述的识别待检测货物数量的方法，其特征在于，所述基于所述分割后的正面货箱图像、所述分割后的顶部货箱图像和对应所述顶部货箱图像的空缺层数确定整托待识别货物的数量，具体包括：

基于所述分割后的正面货箱图像确定垂直方向上货箱的最大层数；

基于所述分割后的顶部货箱图像、对应所述顶部货箱图像的空缺层数和所述最大层数确定各顶部货箱图像对应的实际货箱层数；

对所有顶部货箱图像对应的实际货箱层数进行求和，得到整托待识别货物的数量。

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

采集单元，用于采集整托待检测货物的正面图像和顶部图像；

白平衡单元，用于将所述正面图像和所述顶部图像进行白平衡处理，得到白平衡正面图像和白平衡顶部图像；

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

第二分割单元，用于将所述白平衡顶部图像输入第二分割模型，得到分割后的顶部货箱图像和对应所述顶部货箱图像的空缺层数；其中，所述第二分割模型是基于整托货物的样本白平衡顶部图像以及预先在所述样本白平衡顶部图像上标注的各货箱的四角坐标标签和在所述样本白平衡顶部图像上标注的各货箱的空缺层数标签进行训练得到的；

计算单元，用于基于所述分割后的正面货箱图像、所述分割后的顶部货箱图像和对应所述顶部货箱图像的空缺层数确定整托待识别货物的数量。