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

摘要

本发明提供一种融合注意力模型的烟虫目标检测方法，包括：获取烟虫图像，并制作目标检测的训练数据集；构建神经网络模型，所述神经网络模型以ResNet50为CenterNet目标检测算法的主干特征提取网络，加入注意力模型后的残差结构对输入的特征图进行通道注意；使用所述训练数据集对所述神经网络模型的CenterNet目标检测算法进行训练，以获得训练好的网络模型；将待检测烟虫图像输入训练好的所述神经网络模型，以获得烟虫检测结果。本发明能解决现有烟草车间对烟虫的目标识别存在不精准的问题，能提高烟草车间对烟虫防治的预警效率，降低烟厂管理的生产成本。