高级检索+

基于YOLO-TN和无人机影像的育种小麦分蘖数检测

Tiller counting of breeding wheat based on YOLO-TN and drone imagery

  • 摘要: 小麦分蘖数是小麦育种的核心目标性状之一,然而田间育种小麦分蘖数检测中因人工检测耗时长且主观性强,分蘖相互遮挡、环境条件多样,导致检测成本高、精度较低。为实现育种小麦分蘖数的高精度、高通量检测,该研究提出一种基于无人机图像的轻量化分蘖数检测模型YOLO-TN(YOLO for tiller number),通过检测小麦收割后的茎秆孔数量得到小麦的分蘖数量。为提高检测精度及便于集成至硬件设备,该模型增加新的浅层检测头(SmallHead)以增强小麦秸孔密集小目标的提取能力,同时使用GAM(Global attention mechanism)注意力机制减弱背景噪声干扰,并引入RepViTBlock模块减少模型的计算量和参数量使其轻量化。田间试验结果表明,在单株尺度上,模型的平均精度、F1分数分别达到了93.4%、90.2%,比原模型分别提高了2.7、1.6个百分点。模型的预测总分蘖数与人工标注总分蘖数的决定系数为0.92,均方根误差为2.79。在小区尺度上,模型预测总分蘖数与人工田间调查总分蘖数的决定系数为0.74,均方根误差为4.59。同时模型的参数数量和计算复杂度分别为3.48 M和11.3,比YOLO11s分别减少63%和46.9%。改进后的模型具有高精度、轻量化、高通量的特点,能够在田间情境下基于无人机快速大规模获取育种小麦分蘖数,为高效育种表型分析提供强有力的技术支撑。

     

    Abstract: Tillering number, a pivotal agronomic trait in wheat breeding programs, plays a crucial role in determining yield potential. However, traditional field-based tiller counting methods are plagued by inefficiencies, including labor-intensive manual assessments, subjective judgments, and challenges posed by dense tiller distribution, mutual occlusion, and diverse environmental conditions. These limitations not only hinder accurate phenotyping but also result in high operational costs and low reproducibility across trials. Moreover, human error and variability among observers further compromise data reliability. Consequently, the lack of precise, high-throughput tiller quantification impedes genetic progress and optimal management decisions. To address the limitations of manual phenotyping, there is an urgent need for automated, objective, and scalable solutions. Technologies such as image-based phenotyping coupled with machine learning can enhance measurement accuracy and reduce labor demands in breeding and agronomic research. Consequently, this study proposes YOLO-TN (YOLO for Tiller Number), which is a lightweight detection model designed to estimate wheat tiller counts from post-harvest drone imagery by identifying stem cross-sections. During field experiments, drones captured high-resolution images across diverse breeding materials and lighting conditions, which were subsequently standardized to a 640×640 pixel resolution. To optimize the model for dense, small-object detection within resource-constrained environments, the YOLO-TN architecture incorporates three primary enhancements. First, a shallow detection head, referred to as SmallHead, preserves fine-grained features from early-stage maps to mitigate information loss in deeper layers. Second, a global attention mechanism (GAM) is integrated to improve the model's sensitivity to tillering regions. This module dynamically evaluates and jointly models the importance of both channel-wise and spatial dimensions, thereby strengthening the perception of critical targets while suppressing interference from background clutter such as crop residues and weeds. Finally, to achieve a lightweight profile, RepViTBlocks are incorporated into the backbone network. These blocks employ reparameterized convolutions to significantly reduce computational overhead and parameter counts while simultaneously enhancing the joint representation of local details and global contextual information. Field experiments demonstrated the superior performance of the YOLO-TN model on the wheat tiller dataset. Field breeding experiments demonstrated that the YOLO-TN model exhibited exceptional performance on the tiller dataset of breeding wheat. At the individual plant level, YOLO-TN achieved an mAP@0.5 of 93.4% and an F1 score of 90.2%, marking improvements of 2.7 and 1.6 percentage points, respectively, over the YOLO11n. The coefficient of determination between the model’s predicted total tiller count and the manually annotated total tiller count was 0.92, with an root mean square error of 2.79. Additionally, at the plot scale, the coefficient of determination between the model’s predicted total tiller count and the total tiller count obtained from manual field surveys was 0.74, with an root mean square error of 4.59. Meanwhile, the model's parameter count and computational complexity are 3.48M and 11.3, respectively, representing reductions of 63% and 46.9% compared to YOLO11s. The optimized YOLO-TN model exhibits lightweight architecture, high precision, and high-throughput capabilities, enabling efficient tiller number acquisition in field breeding scenarios using drone technology. Its potential for large-scale wheat phenotyping applications underscores its significance in accelerating breeding programs. This study provides a reliable and scalable solution for tiller number assessment, contributing to the advancement of precision agriculture in wheat breeding.

     

/

返回文章
返回