[发明专利]基于Faster RCNN的卷烟爆珠缺陷检测方法、装置以及设备

说明书

本发明公开了一种基于Faster RCNN的卷烟爆珠缺陷检测方法、装置以及设备，本发明的构思在于通过搭建Faster RCNN架构的爆珠缺陷检测模型，在特征提取网络提取的卷烟爆珠待测图像的初始特征图上生成若干锚框，并经由对锚框进行前后景分类处理，获得得到若干待定的缺陷候选框；接着将缺陷候选框映射到初始特征图中并进行感兴趣区域池化处理，得到相应的目标特征图；最后对该目标特征图再按既定策略进行分类，得到缺陷目标框及其对应的缺陷类别置信度，由此便可以确定出输入的卷烟爆珠图像中的爆珠缺陷。本发明充分利用Faster RCNN对小目标检测精度高的优势，能够显著提升爆珠缺陷检测的效率，对应用环境的适应性更佳，从而满足对爆珠缺陷实时检测的精度及速度的严苛要求。

技术领域

本发明涉及烟草加工领域，尤其涉及一种基于Faster RCNN的卷烟爆珠缺陷检测方法、装置以及设备。

背景技术

爆珠烟因其口味繁多，可以掩盖卷烟烟气的刺激性和提高滤棒截留香气效果等优势而逐渐显示出强劲的增长势头和良好的发展前景，在国内外烟草市场都引起了广泛的关注。由于爆珠在生产过程中容易出现气泡、凹陷、划痕等缺陷，影响卷烟品质，因此对爆珠进行缺陷检测很有必要。

目前针对未包装爆珠的缺陷检测研究相对较少，主要检测方法是基于图像处理技术，先设定若干缺陷类型的判别阈值，通过统计指定色度范围内的像素个数识别拖尾爆珠和空爆珠；通过统计指定灰度范围内的像素个数识别气泡爆珠和斑点爆珠；以及，利用圆形度阈值识别异形爆珠。

但阈值的设定影响因素较多，如噪声，光照强度等，对环境的适应性差；对于小气泡、微小拖尾等不太明显的缺陷尚缺乏有效的检测手段，而且缺陷识别过程中一般会使用到边缘检测技术对从图像中提取出的轮廓进行编码，以识别出气泡爆珠，但其可识别的缺陷类别较为单一，不足以支撑工程应用。

发明内容

由此，本发明旨在提供一种基于Faster RCNN的卷烟爆珠缺陷检测方法、装置以及设备，以及相应地提供了一种数据存储介质以及计算机程序产品，以解决传统爆珠缺陷检测方法适应性差、检测缺陷类别单一，尤其对卷烟爆珠较不明显的缺陷检测效果不理想的问题。

本发明采用的技术方案如下：

第一方面，本发明提供了一种基于Faster RCNN的卷烟爆珠缺陷检测方法，其中包括基于Faster RCNN架构预先构建爆珠缺陷检测模型，所述爆珠缺陷检测模型用于执行如下处理操作：

接收卷烟爆珠待测图像；

利用特征提取网络提取所述卷烟爆珠待测图像的初始特征图；

基于预设的区域推荐网络在所述初始特征图中生成若干锚框；

对所述锚框进行前后景分类，并根据前后景分类结果对所述锚框进行筛选，得到若干缺陷候选框；

将所述缺陷候选框映射到所述初始特征图中并进行感兴趣区域池化处理，得到相应的具有固定尺寸的目标特征图；

基于预设的各缺陷类型的图像特点，对所述目标特征图进行分类，得到缺陷目标框以及对应于所述缺陷目标框的缺陷类别置信度；

根据所述缺陷目标框及所述缺陷类别置信度，确定所述卷烟爆珠待测图像中的爆珠缺陷。

在其中至少一种可能的实现方式中，在所述爆珠缺陷检测模型的训练阶段，将基于区域推荐网络的所述锚框的分类损失、确定所述缺陷候选框的回归损失、基于所述目标特征图的分类损失、确定所述缺陷目标框的回归损失，融合为总损失并作为用于所述爆珠缺陷检测模型进行参数更新的目标函数。

在其中至少一种可能的实现方式中，在所述爆珠缺陷检测模型的训练阶段，采用Adam网络对所述爆珠缺陷检测模型中的连接权重参数进行学习及更新。

在其中至少一种可能的实现方式中，所述特征提取网络采用深度可分离卷积网络。