数据智能

基于PSA-YOLO11n的小麦害虫检测

  • 康继昌 ,
  • 赵连军
展开
  • 山东理工大学,济南 250000

收稿日期: 2025-02-05

  录用日期: 2025-04-21

  网络出版日期: 2025-09-28

基金资助

工业互联网生态智能创新与应用平台(2020SNPT0055)

Wheat Pest Detection Based on PSA-YOLO11n

  • KANG JiChang ,
  • ZHAO LianJun
Expand
  • Shandong University of Technology, Jinan 250000, China
KANG JiCang,E-mail: kangjichang2009@163.com.
ZHAO LianJun, E-mail: lianjunzhao@163.com.

Received date: 2025-02-05

  Accepted date: 2025-04-21

  Online published: 2025-09-28

摘要

针对自然环境中小麦害虫种类繁多、尺寸差异大和生长环境复杂导致检测精度低的问题,提出了一种PSA-YOLO11n小麦害虫检测算法,以提升小麦害虫的检测精度。在 YOLO11n 算法的基础上,对三个关键组成部分进行改进:1)在主干部分引入一种改进空间金字塔池化SimCSPSPPF(Sim CSP Spatial Pyramid Pooling - Fast,SimCSPSPPF) 模块,降低隐藏层的通道数量,加快模型训练速度。2)在中间部分将普通卷积替换为效果更好的感知增强卷积(Perception enhancement convolution,PEC),增强多尺度特征提取能力,提升目标检测速度。 3)回归损失函数更换为AWIoU(Adequate Wise IoU ),改善害虫种类繁多、尺寸差异大造成的检测框失真,提升边界框定位能力。利用IP102数据集进行试验验证,PSA-YOLO11n 与 YOLO11n 相比,mAP提升0.8%,达到89.10%。与Faster R-CNN、RetinaNet、YOLOv5s、YOLOv8n 、YOLOv10n和 YOLO11n 等主流算法进行比较,模型性能均优于其它对比算法。试验结果表明,改进算法PSA-YOLO11n,显著提升了自然环境下多尺度小麦害虫检测精度,为小麦病虫害防治提供一种有效的解决方案。

本文引用格式

康继昌 , 赵连军 . 基于PSA-YOLO11n的小麦害虫检测[J]. 农业大数据学报, 2025 , 7(3) : 294 -306 . DOI: 10.19788/j.issn.2096-6369.000101

Abstract

To address the challenges of low detection accuracy caused by the diverse species, significant size variations, and complex growth environments of wheat pests in natural settings, a PSA-YOLO11n algorithm is proposed to enhance detection precision. Building upon the YOLO11n framework, the proposed improvements include three key components: 1) SimCSPSPPF in Backbone: An improved Spatial Pyramid Pooling-Fast (SPPF) module, SimCSPSPPF, is integrated into the Backbone to reduce the number of channels in the hidden layers, thereby accelerating model training. 2) PEC in Neck: The standard convolution layers in the Neck are replaced with Perception Enhancement Convolutions (PEC) to improve multi-scale feature extraction capabilities, enhancing detection speed. 3) AWIoU Loss Function: The regression loss function is replaced with Adequate Wise IoU (AWIoU), addressing issues of bounding box distortion caused by the diversity in pest species and size variations, thereby improving the precision of bounding box localization. Experimental evaluations on the IP102 dataset demonstrate that PSA-YOLO11n achieves a mean Average Precision (mAP) of 89.10%, surpassing YOLO11n by 0.8%. Comparisons with other mainstream algorithms, including Faster R-CNN, RetinaNet, YOLOv5s, YOLOv8n, YOLOv10n, and YOLO11n, confirm that PSA-YOLO11n outperforms all baselines in terms of detection performance. These results highlight the algorithm’s capability to significantly improve the detection accuracy of multi-scale wheat pests in natural environments, providing an effective solution for pest management in wheat production.

参考文献

[1] HUSSAIN D, ASRAR M, KHALID B, et al. Insect pests of economic importance attacking wheat crop (Triticum aestivum L.) in Punjab, Pakistan. International Journal of Tropical Insect Science, 2022, 42(1):9-20.
[2] ZHANG Q, MEN X, HUI C, et al. Wheat yield losses from pests and pathogens in China. Agriculture, Ecosystems & Environment, 2022, 326:107821.
[3] FAROOK U B, KHAN Z H, AHAD I, et al. A review on insect pest complex of wheat (Triticum aestivum L.). Journal of Entomology and Zoology Studies, 2019, 7(1):1292-1298.
[4] ZHANG M, WANG X, KANG M, et al. A novel agricultural data sharing mode based on rice disease identification. Journal of Agricultural Big Data, 2023, 5(4):13-23.
[5] SABANCI K, ASLAN M F, ROPELEWSKA E, et al. A novel convolutional-recurrent hybrid network for sunn pest-damaged wheat grain detection. Food Analytical Methods, 2022, 15(6):1748-1760.
[6] LI C, CHEN S, MA Y, et al. Wheat pest identification based on deep learning techniques// 2024 IEEE 7th International Conference on Big Data and Artificial Intelligence (BDAI). IEEE, 2024:87-91.
[7] XIAO J, CHEN L, PAN F, et al. Application method affects pesticide efficiency and effectiveness in wheat fields. Pest Management Science, 2020, 76(4):1256-1264.
[8] TUDOR V C, STOICEA P, CHIURCIU I A, et al. The use of fertilizers and pesticides in wheat production in the main European Countries. Sustainability, 2023, 15(4):3038.
[9] ARUN R A, UMAMAHESWARI S. Effective and efficient multi-crop pest detection based on deep learning object detection models. Journal of Intelligent & Fuzzy Systems, 2022, 43(4):5185-5203.
[10] LI R, WANG R, ZHANG J, et al. An effective data augmentation strategy for CNN-based pest localization and recognition in the field. IEEE access, 2019, 7:160274-160283.
[11] GUO B, WANG BB, ZHANG ZH, et al. Improved YOLOv3 crop target detection algorithm. Journal of Agricultural Big Data, 2024, 6(1):40-47.
[12] ZHA M, QIAN W, YI W, et al. A lightweight YOLOv4-Based forestry pest detection method using coordinate attention and feature fusion. Entropy, 2021, 23(12):1587.
[13] LIU J, WANG X, MIAO W, et al. Tomato pest recognition algorithm based on improved YOLOv4. Frontiers in Plant Science, 2022, 13:814681.
[14] YANG S, XING Z, WANG H, et al. Maize-YOLO: a new high- precision and real-time method for maize pest detection. Insects, 2023, 14(3):278.
[15] SUN H, NICHOLAUS I T, FU R, et al. YOLO-FMDI: A lightweight YOLOv8 focusing on a multi-scale feature diffusion interaction neck for tomato pest and disease detection. Electronics, 2024, 13(15):2974.
[16] DONG Q, SUN L, HAN T, et al. PestLite: A novel YOLO-based deep learning technique for crop pest detection. Agriculture, 2024, 14(2):228.
[17] ZHOU X, WEN L, JIE B, et al. Real-time detection algorithm of expanded feed image on the water surface based on improved YOLOv11. Smart Agriculture, 2024, 6(6):155-167.
[18] SONG X, CAO S, ZHANG J, et al. Steel surface defect detection algorithm based on YOLOv8. Electronics, 2024, 13(5):988.
[19] WANG C Y, YEH I H, LIAO H Y M. Yolov9: Learning what you want to learn using programmable gradient information. arXiv preprint arXiv:2402.13616, 2024.
[20] WANG A, CHEN H, LIU L, et al. Yolov10: Real-time end-to-end object detection. arXiv preprint arXiv:2405.14458, 2024.
[21] DU S, ZHANG B, ZHANG P, et al. An improved bounding box regression loss function based on CIOU loss for multi-scale object detection// 2021 IEEE 2nd International Conference on Pattern Recognition and Machine Learning (PRML). IEEE, 2021:92-98.
[22] LI C, LI L, GENG Y, et al. Yolov6 v3. 0: A full-scale reloading. arXiv preprint arXiv:2301.05586, 2023.
[23] SU J, QIN Y, JIA Z, et al. MPE-YOLO: enhanced small target detection in aerial imaging. Scientific Reports, 2024, 14(1):17799.
[24] XUE C, XIA Y, WU M, et al. EL-YOLO: An efficient and lightweight low-altitude aerial objects detector for onboard applications. Expert Systems with Applications, 2024, 256:124848.
[25] WU X, ZHAN C, LAI Y K, et al. Ip102: A large-scale benchmark dataset for insect pest recognition// Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2019:8787-8796.
[26] GEVORGYAN Z. SIoU loss: More powerful learning for bounding box regression. arXiv preprint arXiv:2205.12740, 2022.
[27] YANG Z, WANG X, LI J. EIoU:an improved vehicle detection algorithm based on vehiclenet neural network// Journal of physics: Conference series. IOP Publishing, 2021, 1924(1):012001.
[28] TONG Z, CHEN Y, XU Z, et al. Wise-IoU: Bounding box regression loss with dynamic focusing mechanism. arXiv preprint arXiv:2301.10051, 2023.
[29] QIN H, WANG J, MAO X, et al. An improved faster R-CNN method for landslide detection in remote sensing images. Journal of Geovisualization and Spatial Analysis, 2024, 8(1):2.
[30] YANG Z, LIU Y. A steel surface defect detection method based on improved RetinaNet. Scientific Reports, 2025, 15(1):6045.
[31] WANG D, HE D. Channel pruned YOLO V5s-based deep learning approach for rapid and accurate apple fruitlet detection before fruit thinning. Biosystems Engineering, 2021, 210:271-281.
[32] LIU C, SUN Y, YANG J, et al. Grape recognition and localization method based on 3C-YOLOv8n and depth camera. Smart Agriculture, 2024, 6(6):121-131.
[33] GUAN S, LIN Y, LIN G, et al. Real-time detection and counting of wheat spikes based on improved YOLOv10. Agronomy, 2024, 14(9):1936.
[34] SAPKOTA R, MENG Z, CHURUVIJA M, et al. Comprehensive performance evaluation of YOLOv12, YOLO11, YOLOv10, YOLOv9 and YOLOv8 on detecting and counting fruitlet in complex orchard environments. arXiv:2407.12040, 2024.
文章导航

/