高级检索

      基于改进YOLOv8的农田烟株多光谱图像叶片分割算法

      Leaf Segmentation in Multispectral Images of Farmland Tobacco Plant Based on an Improved YOLOv8 Algorithm

      • 摘要: 高精度的叶片分割与提取是建立基于图像数据的烤烟农情监测算法模型的关键和重要前提,为精准提取农田环境下的烟叶图像信息,本研究建立了可从农田环境下的多光谱图像中精准分割和提取烟株叶片的AO-YOLOv8算法。算法首先对YOLOv8主干网络中的C2f模块进行改进,通过添加聚合注意力机制(Aggregated Attention)模块,实现图像全局信息的完整捕捉,保证叶片分割的完整性;其次,使用GELAN模块替换YOLOv8颈部的C2f模块,并使用重参数化卷积模块OREPA对GELAN进行改进,在减少模型参数量的同时提高了叶片分割精度。试验结果表明,AO-YOLOv8模型在多光谱烟叶叶片实例分割的测试集中,掩膜平均精确度mAP50(Mask)达到87.3%,边界框平均精确度mAP50 (Box)达到89.2%,平均交并比mIoU达到83.4%,相比于YOLOv8模型分别提升了6.99%、8.38%和8.88%,与经典实例分割模型Mask R-CNN和YOLACT相比,分别提升了8.72%、10.26%、7.61%和11.92%、11.64%、17.13%。AO-YOLOv8能够在农田环境下准确分割出多光谱图像中的烟叶叶片,在基于光谱图像数据的烤烟生长发育智能监测研究中具有较好的应用前景。

         

        Abstract: High-precision leaf segmentation and extraction is a crucial prerequisite for establishing flue-cured tobacco agricultural monitoring algorithm models based on image data. To accurately extract tobacco leaf image information in farmland environments, this study developed the AO-YOLOv8 algorithm, which can precisely segment and extract tobacco leaves from multispectral images in farmland environments. The algorithm first enhanced the C2f module in the backbone network of YOLOv8. By adding the Aggregated Attention module, global image information can be captured completely, thereby ensuring the integrity of leaf segmentation. Secondly, the C2f module in the neck of YOLOv8 network was replaced with a GELAN module, which was further refined using the Reparametrized Convolution module (OREPA). This dual modification successfully reduced the number of model parameters while simultaneously improving the leaf segmentation accuracy. The test results showed that the AO-YOLOv8 model achieved 87.3% mAP50(Mask), 89.2% mAP50(Box), and 83.4% mIoU on the multispectral tobacco leaf instance segmentation test set, which represented an improvement of 6.99%, 8.38% and 8.88% over the baseline YOLOv8 model across these three metrics, respectively. Furthermore, compared with the classical instance segmentation models, AO-YOLOv8 showed improvements of 8.72%, 10.26%, and 7.61% over Mask R-CNN, and 11.92%, 11.64%, and 17.13% over YOLACT on the same metrics. AO-YOLOv8 can accurately segment tobacco leaves in multispectral images under farmland environments, making it well-suited for intelligent monitoring of flue-cured tobacco growth and development using spectral image data.

         

      /

      返回文章
      返回