[发明专利]一种货架完整性的检测方法、装置、电子设备及存储介质

权利要求书

1.一种货架完整性的检测方法，包括：

对待检测的目标货架图像进行区域检测，得到所述目标货架图像的目标区域检测信息；所述目标区域检测信息包括目标货架图像中的目标检测框和所述目标检测框所属的目标端部类型；

对所述目标货架图像进行陈列场景分类，得到所述目标货架图像所属的目标陈列场景；

对所述目标货架图像进行直线检测，得到所述目标货架图像中的目标货架层数；

根据所述目标区域检测信息、所述目标陈列场景和所述目标货架层数中的至少一项，确定所述目标货架图像中的货架是否拍摄完整。

2.根据权利要求1所述的方法，其中，所述对待检测的目标货架图像进行区域检测，包括：

通过目标区域检测模型，对待检测的目标货架图像进行区域检测；

所述目标区域检测模型通过如下方式训练得到：

将未标注的第二样本货架图像输入教师区域检测模型，得到第二样本货架图像的伪区域检测信息；所述教师区域检测模型预先采用已标注的第一样本货架图像训练得到；

对所述第一样本货架图像和所述第二样本货架图像进行融合得到，得到第三样本货架图像；所述第三样本货架图像具有标注检测框和标注端部类型；

将所述第三样本货架图像输入学生区域检测模型，得到学生区域检测模型输出的学生检测框和学生端部类型；

根据所述第三样本货架图像的标注检测框和标注端部类型，以及所述学生检测框和所述学生端部类型，对所述学生区域检测模型进行训练，且将训练结果作为所述目标区域检测模型。

3.根据权利要求2所述的方法，其中，所述根据所述第三样本货架图像的标注检测框和标注端部类型，以及所述学生检测框和所述学生端部类型，对所述学生区域检测模型进行训练，且将训练结果作为所述目标区域检测模型，包括：

根据所述第三样本货架图像的标注检测框和所述学生检测框，确定学生区域检测模型的检测框损失；

根据所述第三样本货架图像的标注端部类型和所述学生端部类型，确定学生区域检测模型的端部类型损失；

根据所述学生区域检测模型的检测框损失和所述学生区域检测模型的端部类型损失，对所述学生区域检测模型进行训练，且将训练结果作为所述目标区域检测模型。

4.根据权利要求2所述的方法，还包括：

基于指数移动平均EMA法，根据所述学生区域检测模型中的网络参数，对所述教师区域检测模型中的网络参数进行更新。

5.根据权利要求1所述的方法，其中，所述对所述目标货架图像进行陈列场景分类，包括：

通过目标场景分类模型，对待检测的目标货架图像进行陈列场景分类；

所述目标场景分类模型通过如下方式训练得到：

将未标注的第二样本货架图像输入初始的场景分类模型，得到第二样本货架图像的伪陈列场景；所述初始的场景分类模型预先采用已标注的第一样本货架图像训练得到；

对所述第一样本货架图像和所述第二样本货架图像进行融合得到，得到第三样本货架图像；所述第三样本货架图像具有标注陈列场景；

将所述第三样本货架图像输入初始的场景分类模型，得到第三样本货架图像的预测陈列场景；

根据所述第三样本货架图像的预测陈列场景和所述标注陈列场景对所述初始的场景分类模型进行训练，且将训练结果作为所述目标场景分类模型。

6.根据权利要求1-5中任一项所述的方法，其中，所述根据所述目标区域检测信息、所述目标陈列场景和所述目标货架层数中的至少一项，确定所述目标货架图像中的货架是否拍摄完整，包括：

在根据所述目标区域检测信息确定所述目标货架图像具有底部区域和顶部区域；或者，根据所述目标区域检测信息确定目标货架图像具有顶部区域，且所述目标陈列场景为地龙场景的情况下，确定所述目标货架图像中的货架拍摄完整；

否则，根据所述目标陈列场景和所述目标货架层数，确定目标货架图像中的货架是否拍摄完整。