基于改进的YOLOv5s的花生荚腐病分类
Classification of peanut pod rot based on improved YOLOv5s.
作者信息
Liu Yu, Li Xiukun, Fan Yiming, Liu Lifeng, Shao Limin, Yan Geng, Geng Yuhong, Zhang Yi
机构信息
Hebei Agricultural University, Baoding, China.
State Key Laboratory of North China Crop Improvement and Regulation, Baoding, China.
出版信息
Front Plant Sci. 2024 Apr 15;15:1364185. doi: 10.3389/fpls.2024.1364185. eCollection 2024.
Peanut pod rot is one of the major plant diseases affecting peanut production and quality over China, which causes large productivity losses and is challenging to control. To improve the disease resistance of peanuts, breeding is one significant strategy. Crucial preventative and management measures include grading peanut pod rot and screening high-contributed genes that are highly resistant to pod rot should be carried out. A machine vision-based grading approach for individual cases of peanut pod rot was proposed in this study, which avoids time-consuming, labor-intensive, and inaccurate manual categorization and provides dependable technical assistance for breeding studies and peanut pod rot resistance. The Shuffle Attention module has been added to the YOLOv5s (You Only Look Once version 5 small) feature extraction backbone network to overcome occlusion, overlap, and adhesions in complex backgrounds. Additionally, to reduce missing and false identification of peanut pods, the loss function CIoU (Complete Intersection over Union) was replaced with EIoU (Enhanced Intersection over Union). The recognition results can be further improved by introducing grade classification module, which can read the information from the identified RGB images and output data like numbers of non-rotted and rotten peanut pods, the rotten pod rate, and the pod rot grade. The Precision value of the improved YOLOv5s reached 93.8%, which was 7.8%, 8.4%, and 7.3% higher than YOLOv5s, YOLOv8n, and YOLOv8s, respectively; the mAP (mean Average Precision) value was 92.4%, which increased by 6.7%, 7.7%, and 6.5%, respectively. Improved YOLOv5s has an average improvement of 6.26% over YOLOv5s in terms of recognition accuracy: that was 95.7% for non-rotted peanut pods and 90.8% for rotten peanut pods. This article presented a machine vision- based grade classification method for peanut pod rot, which offered technological guidance for selecting high-quality cultivars with high resistance to pod rot in peanut.
花生荚腐病是影响中国花生产量和品质的主要植物病害之一,会导致大幅减产,且防治具有挑战性。为提高花生的抗病性,育种是一项重要策略。关键的预防和管理措施包括对花生荚腐病进行分级,并筛选出对荚腐病具有高抗性的高贡献基因。本研究提出了一种基于机器视觉的花生荚腐病单例分级方法,该方法避免了耗时、费力且不准确的人工分类,为育种研究和花生荚腐病抗性提供了可靠的技术支持。在YOLOv5s(You Only Look Once版本5小)特征提取主干网络中添加了Shuffle Attention模块以克服复杂背景下的遮挡、重叠和粘连问题。此外,为减少花生荚的漏识别和误识别,将损失函数CIoU(完全交并比)替换为EIoU(增强交并比)。引入等级分类模块可进一步提高识别结果,可以从识别出的RGB图像中读取信息,并输出诸如未腐烂和腐烂花生荚的数量、腐烂荚率和荚腐病等级等数据。改进后的YOLOv5s的Precision值达到93.8%,分别比YOLOv5s、YOLOv8n和YOLOv8s高7.8%、8.4%和7.3%;mAP(平均精度均值)值为92.4%,分别提高了6.7%、7.7%和6.5%。改进后的YOLOv5s在识别准确率方面比YOLOv5s平均提高了6.26%:未腐烂花生荚的识别准确率为95.7%,腐烂花生荚的识别准确率为90.8%。本文提出了一种基于机器视觉的花生荚腐病等级分类方法,为花生中抗荚腐病的优质品种选育提供了技术指导。
相似文献
Front Plant Sci. 2024-4-15
Sensors (Basel). 2023-5-19
Front Plant Sci. 2022-9-28
Comput Struct Biotechnol J. 2023-5-6
Front Plant Sci. 2024-1-11
引用本文的文献
Sci Rep. 2025-5-13
本文引用的文献
Sci Rep. 2021-1-14
IEEE Trans Pattern Anal Mach Intell. 2016-6-6