改进YOLOv5算法的玉米病害检测研究

被引:23
|
作者
苏俊楷
段先华
叶赵兵
机构
[1] 江苏科技大学计算机学院
关键词
玉米病害; YOLOv5; 特征金字塔; 注意力机制; EIOU; 目标检测;
D O I
暂无
中图分类号
TP391.41 []; S435.131 [玉米病害];
学科分类号
080203 ;
摘要
为了解决玉米叶部病害识别技术落后、效率不高、精度不够的问题,提出一种改进的YOLOv5算法识别玉米病害。在保持模型的较低计算量的情况下,同时提升检测速度和算法性能,将传统的YOLOv5s网络的特征提取结构进行改进,在主干网络中添加CA注意力机制,改善目标漏检问题,帮助模型更加精准地定位和识别;在颈部使用BiFPN替代原有的PANet,通过双向的特征融合提升多尺度语义特征的利用,加强对图像深层特征的提取,添加小目标监测层,加强对小物体的检测效果;改进损失函数,引入Focal-EIOU Loss损失函数,提高BBox的回归精度。改进后的YOLOv5算法和传统的YOLOv5s相比,Recall提升了4.61个百分点,平均精度AP上升了4.5个百分点,mAP@0.5提高了2.14个百分点,检测速度上升了4.5 FPS。实验结果表明,改进后的YOLOv5算法在增加极少复杂度的情况下明显提升了算法的效率和性能,并且效果优于传统的YOLOv5s算法。
引用
收藏
页码:933 / 941
页数:9
相关论文
共 7 条
  • [1] 玉米不同部位化学成分、药理作用、利用现状研究进展
    杨小倩
    郅慧
    张辉
    孙佳明
    [J]. 吉林中医药, 2019, 39 (06) : 837 - 840
  • [2] 玉米的品质特性及综合利用研究进展
    马先红
    李峰
    宋荣琦
    [J]. 粮食与油脂, 2019, 32 (01) : 1 - 3
  • [3] Focal and efficient IOU loss for accurate bounding box regression
    Zhang, Yi-Fan
    Ren, Weiqiang
    Zhang, Zhang
    Jia, Zhen
    Wang, Liang
    Tan, Tieniu
    [J]. NEUROCOMPUTING, 2022, 506 (146-157) : 146 - 157
  • [4] EfficientDet: Scalable and efficient object detection[J] . Tan M.,Pang R.,Le Q.V..Proceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition . 2020
  • [5] Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks
    Ren, Shaoqing
    He, Kaiming
    Girshick, Ross
    Sun, Jian
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2017, 39 (06) : 1137 - 1149
  • [6] ImageNet Classification with Deep Convolutional Neural Networks
    Krizhevsky, Alex
    Sutskever, Ilya
    Hinton, Geoffrey E.
    [J]. COMMUNICATIONS OF THE ACM, 2017, 60 (06) : 84 - 90
  • [7] Going deeper with convolutions. Christian Szegedy,Wei Liu,Yangqing Jia,Pierre Sermanet,Scott Reed,Dragomir Anguelov,Dumitru Erhan,Vincent Vanhoucke,Andrew Rabinovich. Proceedings of the 2015 IEEE Conference on Computer Vision and Pattern Recognition . 2015