基于改进YOLOv5注意力模型的农田害虫图像识别

作者:石璐莹; 童顺延; 吴婷; 冯媛; 刘海华
来源:现代信息科技, 2023, 7(10): 70-79.
DOI:10.19850/j.cnki.2096-4706.2023.10.018

摘要

农田害虫防控是一项争分夺秒的挑战,在此过程中害虫种类的正确识别是一项极为重要的环节。针对传统识别害虫检测过程中准确率低、检测目标较小的问题,文章提出了一种基于YOLOv5s和注意力机制的农田害虫图像识别模型。将自注意力机制引入YOLOv5s网络,对上下文信息进行建模,通过建立非局部模型提高网络解决图像远距离和多层次依赖关系的能力。实验结果显示,基于YOLOv5注意力模型的农田害虫图像识别具有较高的检测精度,可以有效识别和定位各类害虫。

全文