中国农业机械化科学研究院集团有限公司 主管

北京卓众出版有限公司 主办

张世豪,董峦,逄正钧,等.基于改进YOLOv5的小麦穗目标检测模型[J].农业工程,2023,13(3):50-56. DOI: 10.19998/j.cnki.2095-1795.2023.03.009
引用本文: 张世豪,董峦,逄正钧,等.基于改进YOLOv5的小麦穗目标检测模型[J].农业工程,2023,13(3):50-56. DOI: 10.19998/j.cnki.2095-1795.2023.03.009
ZHANG Shihao,DONG Luan,PANG Zhengjun,et al.Wheat spike target detection model based on improved YOLOv5[J].Agricultural Engineering,2023,13(3):50-56. DOI: 10.19998/j.cnki.2095-1795.2023.03.009
Citation: ZHANG Shihao,DONG Luan,PANG Zhengjun,et al.Wheat spike target detection model based on improved YOLOv5[J].Agricultural Engineering,2023,13(3):50-56. DOI: 10.19998/j.cnki.2095-1795.2023.03.009

基于改进YOLOv5的小麦穗目标检测模型

Wheat Spike Target Detection Model Based on Improved YOLOv5

  • 摘要: 小麦穗的自动检测在小麦估产和育种方面具有较大科研价值,当前小麦穗检测方面仍存在模型复杂度较高、精度较低等问题。将深度学习技术应用于小麦穗检测,提出了基于改进YOLOv5的小麦穗精确检测模型。模型将YOLOv5主干网络中的卷积模块替换为Ghost卷积,实现模型轻量化;使用ACON激活函数替换默认的SiLU激活函数,从而使激活函数更加灵活以扩大设计空间;使用对所有IoU Loss增加α幂的Alpha-IoU Loss替换YOLOv5默认的CIoU Loss以提高模型前期收敛速度;在网络中加入加权双向特征金字塔(BiFPN),改进的模型可实现参数量降低63.3%、计算量降低66.8%的情况下mAP仅降低2.17%,可满足实际应用和移动端部署的要求。提出了使用解耦头(Decouple Head)替换默认YOLO Head,比官方YOLOv5的mAP提高1.83%,证明了解耦头可以提高模型精度。

     

    Abstract: Automatic detection of wheat spike is of great scientific value in wheat yield estimation and breeding, however there are still problems in current wheat spike detection such as high model complexity and low accuracy.Deep learning technology was applied to wheat spike detection, and an improved YOLOv5 model for accurate detection of wheat spikes was proposed.The model replaced convolution module in the YOLOv5 backbone network with Ghost convolution to achieve a lightweight model; used ACON activation function to replace default SiLU activation function, thus making activation function more flexible to expand design space; used the Alpha-IoU Loss, which increased alpha power for all IoU Losses, to replace default CIoU Loss of YOLOv5.Finally, by adding a weighted bi-directional feature pyramid(BiFPN)to the network, improved model achieved a 63.3% reduction in number of parameters and a 66.8% reduction in computation with only a 2.17% reduction in mAP, met requirements of practical applications and mobile deployments.Use of Decouple Head was proposed to replace default YOLO Head, which improved mAP by 1.83% over official YOLOv5, demonstrated that decoupling head ccould improve model accuracy.

     

/

返回文章
返回