[发明专利]一种桶状物检测方法、装置、电子设备及存储介质

说明书

本发明公开了一种桶状物检测方法、装置、电子设备及存储介质，将待检测图像输入预先训练完成的桶状物检测模型，基于桶状物检测模型对待检测图像进行特征提取得到多通道的第一特征图组；将第一特征图组按照通道数进行分组，通过采用不同尺度的卷积核对不同组内的特征图进行卷积处理得到第二特征图组。使得桶状物检测模型更集中于待检测图像中的细节特征，最后根据所述第二特征图组检测所述待检测图像中的桶状物。从而使得在复杂的场景中，能够准确地检测出待检测图像中的桶状物。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种桶状物检测方法、装置、电子设备及存储介质。

背景技术

随着国内经济的迅速发展，国民的消费水平也在不断提高，越来越多的人通过饭店餐馆外卖来解决生活饮食，在餐饮业高消费的背景下，餐饮后厨的规范化管理问题也成为重中之重，很多外表光鲜亮丽的饭店，其后厨却脏乱不堪，尤其是垃圾桶的开盖状态，若垃圾桶由于满溢或者未对其进行关闭而开放垃圾桶桶盖，将导致苍蝇横飞，加速与异味细菌的传播。因此，需要对后厨垃圾桶桶盖的开盖状态进行规定，比如必须及时关闭垃圾桶，垃圾桶满溢后必须及时清理垃圾，若无法做到有效管理，将严重威胁着厨房的卫生安全，甚至会对食客的身体健康产生巨大的不良影响。另外，对于小区内或景区内垃圾桶溢满或垃圾桶附近脏乱差同样会影响住户或游客的观感体验。

为了解决上述问题，首先需要检测出环境中的桶状物(例如垃圾桶)，但是由于场景背景复杂，现有技术中检测桶状物的准确性较差，如何在场景中准确地检测到桶状物是目前亟需解决的问题。

发明内容

本发明实施例提供了一种桶状物检测方法、装置、电子设备及存储介质，用以解决现有技术中检测桶状物的准确性较差的问题。

本发明实施例提供了一种桶状物检测方法，所述方法包括：

获取待检测图像，将所述待检测图像输入预先训练完成的桶状物检测模型；

基于所述桶状物检测模型，对所述待检测图像进行特征提取得到多通道的第一特征图组；将所述第一特征图组按照通道数进行分组，采用不同尺度的卷积核对不同组内的特征图进行卷积处理得到第二特征图组；根据所述第二特征图组检测所述待检测图像中的桶状物。

进一步地，所述将卷积处理后的特征图进行拼接得到第二特征图组包括：

将卷积处理后的特征图进行拼接得到的特征图组作为第三特征图组；

将所述第三特征图组按照通道数进行分组，将每组内的特征图与预先训练得到的权重参数相乘得到第二特征图组。

进一步地，所述方法还包括：

将根据所述第二特征图组检测出的所述待检测图像中的桶状物区域图像作为疑似区域图像；

将所述疑似区域图像输入预先训练完成的桶状物分类模型，基于所述桶状物分类模型中的第一分支网络结构，检测所述疑似区域图像中的物体是否为桶状物。

进一步地，所述方法还包括：

若基于所述桶状物分类模型中的第一分支网络结构，检测所述疑似区域图像中的物体为桶状物，基于所述桶状物分类模型中的第二分支网络结构，检测所述桶状物顶盖的开关状态。

进一步地，所述方法还包括：

若基于所述桶状物分类模型中的第二分支网络结构，检测所述桶状物顶盖为打开状态，将所述待检测图像作为候选报警图像；当确定出设定数量的连续帧候选报警图像，输出报警提示信息。

进一步地，所述基于所述桶状物检测模型，对所述待检测图像进行特征提取得到多通道的第一特征图组；将所述第一特征图组按照通道数进行分组，采用不同尺度的卷积核对不同组内的特征图进行卷积处理得到第二特征图组；根据所述第二特征图组检测所述待检测图像中的桶状物包括：