吴利刚,陈乐,刘泽鹏,等. 基于YOLOv8-ABW的黄花成熟度检测方法[J]. 农业工程学报,2024,40(13):262-272. DOI: 10.11975/j.issn.1002-6819.202401074
    引用本文: 吴利刚,陈乐,刘泽鹏,等. 基于YOLOv8-ABW的黄花成熟度检测方法[J]. 农业工程学报,2024,40(13):262-272. DOI: 10.11975/j.issn.1002-6819.202401074
    WU Ligang, CHEN Le, LIU Zepeng, et al. YOLOv8-ABW based method for detecting Hemerocallis citrina Baroni maturity[J]. Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 2024, 40(13): 262-272. DOI: 10.11975/j.issn.1002-6819.202401074
    Citation: WU Ligang, CHEN Le, LIU Zepeng, et al. YOLOv8-ABW based method for detecting Hemerocallis citrina Baroni maturity[J]. Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 2024, 40(13): 262-272. DOI: 10.11975/j.issn.1002-6819.202401074

    基于YOLOv8-ABW的黄花成熟度检测方法

    YOLOv8-ABW based method for detecting Hemerocallis citrina Baroni maturity

    • 摘要: 为实现黄花成熟度的快速、高精度识别,针对其相似特征识别精确度低以及相互遮挡检测困难的问题,提出一种基于YOLOv8-ABW的黄花成熟度检测方法。该研究在特征提取网络中加入结合注意力机制的尺度特征交互机制(attention based intra-scale feature interaction, AIFI),更好地提取黄花特征信息,提高检测的精确度。在特征融合网络中,进一步采用加权的双向特征金字塔特征融合网络(bidirectional feature pyramid network, Bi FPN),实现更高层次的跨通道特征融合,有效减少通道中的特征冗余。此外使用WIoUv3作为损失函数,聚焦普通质量的锚框,提高模型的定位性能。试验结果表明:YOLOv8-ABW模型检测精确度为82.32%,召回率为83.71%,平均精度均值mAP@0.5和mAP@0.5:0.95分别为88.44%和74.84%,调和均值提升至0.86,实时检测速度为214.5帧/s。与YOLOv8相比,YOLOv8-ABW的精确度提高1.41个百分点,召回率提高0.75个百分点,mAP@0.5和mAP@0.5:0.95分别提升1.54个百分点和1.42个百分点。对比RT-DETR、YOLOv3、YOLOv5、YOLOv7模型,YOLOv8-ABW参数量最少,仅为3.65×106,且模型浮点运算量比YOLOv7少96.3 G。体现出YOLOv8-ABW 模型能够在黄花成熟度检测中平衡检测精确度和检测速度,综合性能最佳,为黄花智能化实时采摘研究提供技术支持。

       

      Abstract: Here, rapid and high-precision detection was proposed for Hemerocallis citrina Baroni maturity using YOLO v8-ABW. Current challenges were overcome on the similar features and mutual occlusions during recognition. The precision and efficiency of detection were improved to provide crucial technical support to intelligent real-time harvesting. Feature extraction and utilization were also greatly improved in the detection process of Hemerocallis citrina Baroni maturity. Attention-based Intra-scale Feature Interaction (AIFI) was incorporated into the feature extraction network. Feature information was interacted with and combined from various scales. The detection precision was enhanced to more effectively extract the information about Hemerocallis citrina Baroni. Specifically, the AIFI was used to weigh the features using the attention mechanism. More key areas were focused on extracting the features, in order to reduce the interference of noise and redundancy. Meanwhile, the scale feature interaction was used to fully utilize the feature information of different scales, thereby enhancing the precision and robustness of feature extraction. A weighted bidirectional feature pyramid feature fusion network (BiFPN) was used in the feature fusion network. This network structure was achieved in the complementarity and enhancement of various layers of feature information using cross-channel feature fusion. Compared with the traditional Feature Pyramid Network, BiFPN was used to retain more original feature information and fuse the features of different layers in a weighted manner, thus enriching the fused features. In addition, Bi FPN had effectively reduced the feature redundancy in the channels, thereby enhancing the speed and efficiency of detection. Moreover, WIoUv3 was used in the selection of the loss function. The loss function was specifically optimized for the standard quality anchor frames, in order to focus more on the location of targets during training. The WIoUv3 loss function was introduced to successfully enhance the localization performance, leading to more accurate and reliable detection. The experiment validated that the YOLOv8-ABW model was performed better to detect the maturity of Hemerocallis citrina Baroni. The precision reached 82.32%, the recall was 83.71%, mAP@0.5 and mAP@0.5:0.95 were 88.44% and 74.84%, respectively. The harmonic mean was improved to 0.86, and the real-time detection speed even reached 214.5 frames per second (f/s). Compared with the original one, the YOLOv8-ABW model showed improvements in the precision, recall, mAP@0.5, and mAP@0.5:0.95 by 1.41, 0.75, 1.54, and 1.42 percentage points, respectively. Compared to the rest, YOLOv8-ABW shared the least number of parameters, only 3.65×106. The floating-point operations were 96.3 G less than the YOLOv7. The high precision YOLOv8-ABW also exhibited high computational efficiency for the real-time detection tasks related to Hemerocallis Citrina Baroni maturity. In summary, the YOLOv8-ABW model demonstrated excellent performance for the detection. The simplified complexity was also obtained with the detection precision and speed. The finding can offer robust technical support to the intelligent real-time harvesting of Hemerocallis citrina Baroni.

       

    /

    返回文章
    返回