高级检索+

基于改进Swin Transformer与融合无人机多光谱图像的小麦幼苗识别方法

Wheat seedling recognition method based on improved swin transformer with UAV multispectral image fusion

  • 摘要: 为解决传统小麦幼苗监测费时费力、边界分割模糊及多光谱波段利用不足的问题,实现复杂田间条件下小麦幼苗的高精度、高时效识别,该研究提出改进Swin Transformer与融合无人机多光谱图像的小麦幼苗语义分割框架—多光谱混合网络(multi-spectral hybrid network, MS-HybridNet)。模型以Swin Transformer为基线架构,适配无人机5通道多光谱影像输入,构建可见光-红边(red edge, RE)-近红外(near infrared, NIR)波段协同机制,利用红边波段对叶绿素的敏感性和近红外区分作物与土壤的优势,增强模型对小麦幼苗与复杂田间背景之间的区分能力。构建CNN-Transformer混合编码器,结合深度卷积神经网络(convolutional neural network, CNN)提取局部边缘细节,Transformer捕捉全局空间分布的双重特性,完整保留作物形态信息。在瓶颈层引入空洞空间金字塔池化(atrous spatial pyramid pooling, ASPP)模块,通过多尺度空洞卷积扩大感受野,增强对不同尺寸幼苗的适配能力。设计带深度监督的解码器结构,重点强化小目标幼苗和粘连叶片的边界分割精度。在小麦幼苗数据集上的对比试验结果表明,MS-HybridNet的平均像素精度、召回率、F1分数、平均交并比和边界交并比分别达到84.1%、73.4%、76.0%、80.5%和71.0%,较基线Swin Transformer分别提升5.0、2.3、2.1、6.5和5.9个百分点,参数量为27.36 M,单张影像推理时间仅18.2 ms,满足无人机航拍数据的实时处理需求。在不同光谱输入模态的对比试验表明,可见光-红边-近红外三波段协同输入时较单可见光输入的平均像素精度、召回率、F1分数、平均交并比和边界交并比分别提升12.0、1.9、0.2、18.2和10.3个百分点。该研究可为精准农业变量施肥与精细化田间管理提供一定的技术支撑。

     

    Abstract: Accurate and efficient wheat seedling monitoring is a critical prerequisite for precision variable-rate fertilization and refined field management, directly affecting crop yield formation and agricultural production efficiency. Traditional manual field survey methods suffer from low efficiency, limited spatial coverage and high labor costs, while conventional visible-light deep learning models face persistent challenges including severe spectral confusion between seedlings and stubble or weeds in low-coverage seedling scenarios, blurred boundary segmentation under leaf adhesion conditions, and insufficient utilization of vegetation-sensitive spectral bands. To address these issues, this study proposes the Multi-Spectral Hybrid Network (MS-HybridNet), a semantic segmentation framework integrating an improved Swin Transformer with unmanned aerial vehicle (UAV) multispectral imagery, to achieve high-precision wheat seedling identification under complex field conditions. The proposed model takes Swin Transformer as the baseline architecture and is adapted to 5-channel UAV multispectral image input covering blue, green, red, red edge and near-infrared bands. It constructs a visible-red edge-near infrared band synergy mechanism, leveraging the chlorophyll sensitivity of the 730 nm red edge band and the crop-soil differentiation advantage of the 840 nm near-infrared band to enhance spectral discrimination between wheat seedlings and complex backgrounds including stubble, weeds and bare soil. A CNN-Transformer hybrid encoder is designed to replace the original pure Transformer encoding structure, combining the local edge detail extraction capability of convolutional layers and the global spatial distribution modeling advantage of self-attention mechanisms to fully preserve crop morphological features and alleviate segmentation confusion in leaf adhesion areas. An Atrous Spatial Pyramid Pooling module is introduced at the bottleneck layer to expand the receptive field through multi-scale dilated convolutions, strengthening feature adaptability to seedlings of different sizes, and a decoder with deep supervision is adopted to supplement multi-scale auxiliary supervision signals to enhance the boundary segmentation accuracy of small-target seedlings and adherent leaves. Experimental results on the self-built wheat seedling multispectral dataset demonstrate that MS-HybridNet achieves 84.1% mean pixel accuracy (mPA), 73.4% recall, 76.0% F1-Score, 80.5% mean Intersection over Union (mIoU), and 71.0% Boundary IoU (bIoU) on the test set. Compared with the baseline Swin Transformer, mPA, recall, F1-Score, mIoU and bIoU are improved by 5.0, 2.3, 2.1, 6.5 and 5.9 percentage points respectively. Compared with Mask2Former with the closest segmentation accuracy, mIoU increases by 5.2 percentage points, with 39.4% fewer parameters and 28.9% faster inference speed at only 18.2 ms per image. Compared with ResNet-UNet and DeepLabV3+, mIoU is improved by 14.2 and 18.4 percentage points, and bIoU rises by 5.7 and 10.4 percentage points respectively. Spectral ablation experiments further verify that the three-band synergistic input improves mIoU and bIoU by 18.2 and 10.3 percentage points respectively compared with single RGB input. The proposed method maintains controllable computational overhead while significantly improving segmentation accuracy, fully meeting the real-time processing requirements of UAV aerial survey data and providing reliable technical support for variable-rate fertilization and refined field management in precision agriculture.

     

/

返回文章
返回