计算机工程

作者：候瑞环, 杨喜旺, 王智超, 高佳鑫

单位：中北大学 大数据学院, 太原 030051

关键词：林业害虫检测,YOLOv4模型,深度学习,三分支注意力,Focal loss函数,加权双向特征金字塔网络

针对现有基于深度学习的林业昆虫图像检测方法存在检测精度低和检测速度慢的问题，提出一种结合改进PANet结构与三分支注意力机制的目标检测方法YOLOv4-TIA。通过对样本数量较少的昆虫类别进行数据增强，实现样本均衡分布。利用三分支注意力机制改进YOLOv4中的CSPDarkNet53骨干网络，同时通过旋转操作和残差变换建立维度间的依存关系，以提高有效的特征通道权重，在PANet结构上增加将跳跃连接与跨尺度连接相结合的特征融合方式，从而获取更丰富的语义信息和位置信息。在此基础上，采用Focal loss函数优化分类损失，解决正负样本不均衡的问题。实验结果表明，该方法的精确率和召回率分别达到85.9%和91.2%，相比SSD、Faster R-CNN、YOLOv4方法，其在保证检测速度的同时，能够有效提高检测精度，且实现对林业害虫的实时精确监测。

来源：2022年第4期

《计算机工程》期刊编辑部

查看计算机工程杂志2022年第4期