一种融合注意力模型的烟虫目标检测方法与流程

1.本发明涉及烟虫检测的技术领域，尤其涉及一种融合注意力模型的烟虫目标检测方法。


背景技术：

2.在烟草生产加工环节，生产车间需要常年保持恒温恒湿的环境，物料的进入、人员的进出、各种车辆的进出等，均不可避免的会产生烟草害虫，并长期在车间隐匿和繁殖，严重地影响到卷烟品质。为有效降低烟虫对卷烟品质的影响，对于烟虫的防治和治理工作十分必要和需要。为此对烟虫识别采用的方法有多种多样，其中采用图像处理技术对粘虫板上的烟虫进行处理和预警，但其缺点是当粘虫板上有较多烟虫或者存在烟末时则无法实现烟虫的精准检测识别。目标检测是计算机视觉领域中的重要研究课题之一，它的任务就是定位出图像中目标的位置并识别目标的类别。这项技术广泛应用于人脸检测、缺陷检测、自动驾驶、智能监控等，并拥有非常广阔的发展前景。目前主流的目标检测算法都是基于深度学习，比如rcnn，fasterrcnn，yolo等。在实际的目标检测场景中也存在很多不利因素，例如光照变化、目标形状的变化等，尤其是当被检测目标过小时，神经网络对于小目标难以提取有效特征信息，严重影响了目标检测算法的精确度。因此，如何提供一种目标检测方法对烟虫进行精准识别，具有重要的研究意义。


技术实现要素：

3.本发明提供一种融合注意力模型的烟虫目标检测方法，解决现有烟草车间对烟虫的目标识别存在不精准的问题，能提高烟草车间对烟虫防治的预警效率，降低烟厂管理的生产成本。
4.为实现以上目的，本发明提供以下技术方案：
5.一种融合注意力模型的烟虫目标检测方法，包括：
6.获取烟虫图像，并制作目标检测的训练数据集；
7.构建神经网络模型，所述神经网络模型以resnet50为centernet目标检测算法的主干特征提取网络，加入注意力模型后的残差结构对输入的特征图进行通道注意；
8.使用所述训练数据集对所述神经网络模型的centernet目标检测算法进行训练，以获得训练好的网络模型；
9.将待检测烟虫图像输入训练好的所述神经网络模型，以获得烟虫检测结果。
10.优选的，所述制作目标检测的训练数据集包括：
11.使用不同的图像采集设备采集样本图像，再对所述样本图像进行数据增强处理，包括裁剪、平移、亮度和高斯噪声；
12.对所述样本图像进行数据集标注，并生成对应的xml文件。
13.优选的，所述对所述样本图像进行数据集标注，包括：
14.打开图片目录，选择标注文件保存地址，创建边界框，锁定目标范围和选择目标类
型，最后将数据集整理为voc格式。
15.优选的，所述制作目标检测的训练数据集还包括：
16.对所述样本图像进行图像质量评估，并根据所述样本图像的质量评估值对烟虫图像进行分类，以形成不同的训练数据集。
17.优选的，所述神经网络模型以resnet50为centernet目标检测算法的主干特征提取网络，包括：
18.对centernet网络进行改进，将resnet50深度残差神经网络作为主干特征提取网络，所述resnet
‑
50深度残差神经网络包括：输入层，最大池化层和conv block和identity block。
19.优选的，所述加入注意力模型后的残差结构对输入的特征图进行通道注意，包括：
20.将注意力模型插入到resnet50网络中残差结构的主干末尾部分；
21.注意力模型首先对输入的特征图进行全局平均池化，然后通过自适应大小的一维卷积生成通道特征注意力；
22.使用非线性sigmoid函数生成通道权重，最后将生成的每个通道权重与输入特征图对应元素相乘，获得通道注意后的特征图。
23.优选的，所述将注意力模型插入到resnet50网络中残差结构的主干末尾部分，包括：
24.通过在resnet50网络中的conv block和identity block的主干末尾部分加入注意力模型，以对输入注意力模型的特征图进行通道注意。
25.优选的，所述将待检测烟虫图像输入训练好的所述神经网络模型，以获得烟虫检测结果，包括：
26.利用主干特征提取网络resnet50对输入的图像提取特征，对提取的特征利用反卷积进行上采样得到更高分辨率特征图，然后分别进行热力图预测、中心点预测和宽高预测，最后输出检测结果。
27.优选的，所述对输入的特征图进行通道注意，包括：
28.其中，m
c
(f)为通道注意力，f表示输入特征图，avgpool(f)表示全局平均池化，表示大小为k的一维卷积，σ表示sigmoid函数。
29.本发明提供一种融合注意力模型的烟虫目标检测方法，通过构建神经网络模型以resnet50为centernet目标检测算法的主干特征提取网络，加入注意力模型后的残差结构对输入的特征图进行通道注意，并将待检测烟虫图像输入训练好的所述神经网络模型，以获得烟虫检测结果。能解决现有烟草车间对烟虫的目标识别存在不精准的问题，能提高烟草车间对烟虫防治的预警效率，降低烟厂管理的生产成本。
附图说明
30.为了更清楚地说明本发明的具体实施例，下面将对实施例中所需要使用的附图作简单地介绍。
31.图1是本发明提供的一种融合注意力模型的烟虫目标检测方法的示意图。
32.图2是本发明提供的centernet主干特征提取网络部分融合注意力模型后的结构
示意图。
33.图3是本发明提供的注意力模型对输入特征图的处理流程图。
具体实施方式
34.为了使本技术领域的人员更好地理解本发明实施例的方案，下面结合附图和实施方式对本发明实施例作进一步的详细说明。
35.针对烟草车间的烟虫识别不精准的问题，本发明提供一种融合注意力模型的烟虫目标检测方法，通过构建神经网络模型以resnet50为centernet目标检测算法的主干特征提取网络，加入注意力模型后的残差结构对输入的特征图进行通道注意，并将待检测烟虫图像输入训练好的所述神经网络模型，以获得烟虫检测结果。解决现有烟草车间对烟虫的目标识别存在不精准的问题，能提高烟草车间对烟虫防治的预警效率，降低烟厂管理的生产成本。
36.如图1所示，一种融合注意力模型的烟虫目标检测方法，包括：
37.s1：获取烟虫图像，并制作目标检测的训练数据集。
38.s2：构建神经网络模型，所述神经网络模型以resnet50为centernet目标检测算法的主干特征提取网络，加入注意力模型后的残差结构对输入的特征图进行通道注意。
39.s3：使用所述训练数据集对所述神经网络模型的centernet目标检测算法进行训练，以获得训练好的网络模型。
40.s4：将待检测烟虫图像输入训练好的所述神经网络模型，以获得烟虫检测结果。
41.具体地，视觉注意力机制是人类视觉所特有的大脑信号处理机制，人类视觉通过快速扫描全局图像，获得需要重点关注的感兴趣区域，也就是一般所说的注意力焦点，而后对这一区域投入更多注意力资源，以获取更多所需要关注目标的细节信息，而抑制其他无用信息。为解决目标特征信息利用不充分的问题，使神经网络把更多的注意力放在对目标表达更有效的特征信息上，可将注意力模型引入resnet50网络内，如图2所示，使特征提取网络对烟虫图像中的目标特征提取更为精准，进而能更精准识别烟虫，以对烟虫进行防治，能降低烟厂管理的生产成本。
42.进一步，所述制作目标检测的训练数据集，包括：
43.使用不同的图像采集设备采集样本图像，再对所述样本图像进行数据增强处理，包括裁剪、平移、亮度和高斯噪声。对所述样本图像进行数据集标注，并生成对应的xml文件。
44.更进一步，所述对所述样本图像进行数据集标注，包括：打开图片目录，选择标注文件保存地址，创建边界框，锁定目标范围和选择目标类型，最后将数据集整理为voc格式。
45.进一步，所述制作目标检测的训练数据集，还包括：
46.对所述样本图像进行图像质量评估，并根据所述样本图像的质量评估值对烟虫图像进行分类，以形成不同的训练数据集。
47.在实际应用中，通过对样本图像的图像质量评估，可获得高质量样本图像，同时，也对相似样本与不相似样本进行分类，进而对后期的对图像特征提取模型进行模型训练，利用图像的质量评估值对模型训练过程进行约束，使得训练好的图像特征提取模型不仅能够学习到图像特征提取的能力，还能够降低图像质量差的随机变换图像对特征提取结果的
影响，提高了模型图像特征学习的效果。
48.进一步，所述神经网络模型以resnet50为centernet目标检测算法的主干特征提取网络，包括：
49.对centernet网络进行改进，将resnet50深度残差神经网络作为主干特征提取网络，所述resnet
‑
50深度残差神经网络包括：输入层，最大池化层和conv block和identity block。
50.如图2和图3所示，所述加入注意力模型后的残差结构对输入的特征图进行通道注意，包括：
51.将注意力模型插入到resnet50网络中残差结构的主干末尾部分。
52.注意力模型首先对输入的特征图进行全局平均池化，然后通过自适应大小的一维卷积生成通道特征注意力。
53.使用非线性sigmoid函数生成通道权重，最后将生成的每个通道权重与输入特征图对应元素相乘，获得通道注意后的特征图。
54.进一步，所述将注意力模型插入到resnet50网络中残差结构的主干末尾部分，包括：
55.通过在resnet50网络中的conv block和identity block的主干末尾部分加入注意力模型，以对输入注意力模型的特征图进行通道注意。
56.具体地，如图2所示，在resnet50网络中增加注意力模型，使resnet50网络中，对图像特征提取中增加了通道注意力，能更为有效对特征提取的精准度。
57.如图2所示，所述将待检测烟虫图像输入训练好的所述神经网络模型，以获得烟虫检测结果，包括：
58.利用主干特征提取网络resnet50对输入的图像提取特征，对提取的特征利用反卷积进行上采样得到更高分辨率特征图，然后分别进行热力图预测、中心点预测和宽高预测，最后输出检测结果。
59.所述对输入的特征图进行通道注意，包括：其中，m
c
(f)为通道注意力，f表示输入特征图，avgpool(f)表示全局平均池化，表示大小为k的一维卷积，σ表示sigmoid函数。
60.可见，本发明提供一种融合注意力模型的烟虫目标检测方法，通过构建神经网络模型以resnet50为centernet目标检测算法的主干特征提取网络，加入注意力模型后的残差结构对输入的特征图进行通道注意，并将待检测烟虫图像输入训练好的所述神经网络模型，以获得烟虫检测结果。能解决现有烟草车间对烟虫的目标识别存在不精准的问题，能提高烟草车间对烟虫防治的预警效率，降低烟厂管理的生产成本。
61.以上依据图示所示的实施例详细说明了本发明的构造、特征及作用效果，以上所述仅为本发明的较佳实施例，但本发明不以图面所示限定实施范围，凡是依照本发明的构想所作的改变，或修改为等同变化的等效实施例，仍未超出说明书与图示所涵盖的精神时，均应在本发明的保护范围内。