张东彦,高玥,程涛,等. 基于YOLOv8-FECA的密集场景下小麦赤霉病孢子目标检测[J]. 农业工程学报,2024,40(21):1-11. DOI: 10.11975/j.issn.1002-6819.20240614921
    引用本文: 张东彦,高玥,程涛,等. 基于YOLOv8-FECA的密集场景下小麦赤霉病孢子目标检测[J]. 农业工程学报,2024,40(21):1-11. DOI: 10.11975/j.issn.1002-6819.20240614921
    ZHANG DongYan, GAO Yue, CHENG Tao, et al. Detection of wheat scab spores in dense scene based on YOLOv8-FECA[J]. Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 2024, 40(21): 1-11. DOI: 10.11975/j.issn.1002-6819.20240614921
    Citation: ZHANG DongYan, GAO Yue, CHENG Tao, et al. Detection of wheat scab spores in dense scene based on YOLOv8-FECA[J]. Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 2024, 40(21): 1-11. DOI: 10.11975/j.issn.1002-6819.20240614921

    基于YOLOv8-FECA的密集场景下小麦赤霉病孢子目标检测

    Detection of wheat scab spores in dense scene based on YOLOv8-FECA

    • 摘要: 针对小麦赤霉病孢子图像中密集分布小目标的漏检错检问题,该研究设计了一种针对该场景下小目标检测模型YOLOv8-FECA。以YOLOv8为基准模型,先添加新的小目标检测层,以此增强网络对更小目标语义信息的捕捉以及提高其特征描述的准确性;其次,构建融合特征的焦点注意力机制(focal efficient channel attention module,FECA)以实现对小目标和密集分布目标的关注;再次,引入Wise-IoU Loss与DFL Loss结合作为边界框的回归损失,提高模型的收敛能力以及对边界框预测的准确性;最后,在不同密集场景和不同光线环境下验证了模型的鲁棒性。结果表明,优化后模型YOLOv8-FECA相比YOLOv8在孢子数据集中的平均精度均值mAP@0.5提高了4.3%,达到96.8%,且改进模型的FPS(frames per second)也达到实时检测要求。该研究设计的YOLOv8-FECA网络模型在保持高性能的同时能够有效提升密集场景小目标检测的精度,可为其他作物病害孢子的检测提供了新的思路。

       

      Abstract: Wheat scab represents a significant threat to global wheat production, with the primary causal agents being the fungi Fusarium graminearum and Fusarium asiaticum. The dissemination of Fusarium spores represents a pivotal step in the transmission and infection of wheat scab. Consequently, the early and precise identification of pathogen spore concentrations is of paramount importance for the prompt detection of wheat scab, the safeguarding of wheat yield and quality, and the assurance of food safety. Conventional techniques, such as microscopic observation and medium isolation, are often time-consuming and require sophisticated technical expertise, posing a challenge for rapid response and the management of large-scale crop disease transmission. To address the issue of missed and erroneous detections of densely distributed small targets in scab spore images, this study has developed a small target detection model, YOLOv8-FECA. In this study, a small target detection layer with a feature map size of 160×160 was first added to the original YOLOv8 baseline model. This enhancement aimed to improve the network's ability to capture the semantic information of small targets and to enhance the accuracy of feature description. Secondly, a focus attention mechanism, FECA, was designed by combining the advantages of the CBAM and ECA modules. This module was added at the neck end of the network to reduce information loss during the downsampling process. Ultimately, in order to enhance the model's convergence capability and the precision of bounding box estimation, Wise-IoU Loss and DFL Loss were integrated as the regression loss for the bounding box. The experimental results demonstrate that Compared to the original YOLOv8 baseline model, The small target detection layer was added, the focus attention mechanism was introduced, and Wise-IoU Loss and DFL were introduced After Loss improvement, the average detection accuracy of the new YOLOv8-FECA on spore data set mAP@.5 and mAP@.5:.95 were respectively increased by 4.3% and 6.3% compared with the benchmark model Yolov8n. In terms of performance, the YOLOv8-FECA model has demonstrated an improvement in both accuracy and recall in comparison to YOLOv8n. The enhanced model demonstrates an elevated degree of accuracy and comprehensiveness in identifying wheat scab spores, with an increase of 2.6% in Precision and 3.8% in Recall. Although the frame rate per second (FPS) of YOLOv8-FECA has decreased in comparison to YOLOv8n (from 134 to 106), it nevertheless maintains a high detection speed and achieves notable enhancements in other performance indicators. This shows that YOLOv8-FECA successfully improves the detection accuracy while maintaining real-time detection capability, and mAP @ 0.5 reaches 96.8 %. The verification of the model's robustness demonstrated that accurate detection of spores is possible even in challenging scenarios, such as those involving high spore density and low light conditions. Moreover, in comparison with prevalent target detection models, including YOLOv5s, YOLOv7-tiny, SSD network, and the exemplar two-stage Faster-R-CNN, YOLOv8-FECA exhibits superior performance in terms of mAP. The results demonstrate that the improvement strategy is effective and that the new model is superior to existing methods for detecting small targets, such as scab spores. This study provides technical support for the automatic detection of wheat scab spores in the field and the early warning of scab.

       

    /

    返回文章
    返回