高级检索+

MDD-YOLOv11:基于多模态特征融合的番茄成熟度识别方法

MDD-YOLOv11: a tomato maturity recognition method using multimodal feature fusion

  • 摘要: 针对番茄成熟度识别任务中存在背景条件复杂、遮挡、重叠的问题,该研究提出一种基于RGB-NIR(red green blue near-infrared)多模态图像的双主干网络模型MDD-YOLOv11,用来进行番茄成熟度的识别。首先,针对RGB图像易受非均匀光照变化干扰以及NIR图像纹理特征区分度不足的问题,在双主干网络中引入多分支卷积(multi-branch convolution, MConv)模块,该模块通过并行集成多尺度卷积核构建异质感受野特征空间,在RGB分支通过多尺度信息补偿提升光照鲁棒性,在NIR分支则通过细粒度特征挖掘增强模型的特征表达能力。其次,针对遮挡的识别,在Backbone与Neck部分集成跨阶段可变形(C3k2 Block with deformable convolution v4,C3k2_DCNv4)模块,通过自适应调整特征采样位置,有效弥补遮挡导致的信息缺失,提升对遮挡果实的建模能力。最后,针对多模态特征融合中存在的特征对齐与融合冗余问题,在Backbone网络L3、L4不同尺度的特征层设计可变形跨模态注意力融合(deformable cross-modal attention fusion,DCAF)模块,利用可变形卷积实现空间维度的精准对齐,结合跨模态窗口注意力机制抑制冗余并挖掘互补信息,以提升特征融合能力。试验结果表明,所提模型在所选3种背景光混合的数据集上的精确率、召回率和平均精度均值(mean average precision,mAP0.5和mAP0.5:0.95)分别为94.8%、94.5%、98.7%和86.3%,较基准模型YOLOv11n-L3L4分别高出1.7、1.6、1.6和3.4个百分点。将MDD-YOLOv11应用到自建数据集(Hebei Agricultural University Tomato,HBAUT)上,使mAP0.5:0.95达到了83.4%,较基准模型提高2.7个百分点。综上可知,该研究提出的模型在复杂光照和遮挡条件下具有较好的番茄成熟度识别效果,可为温室番茄自动化采摘的视觉感知环节提供技术支持。

     

    Abstract: Tomato maturity recognition is an important visual perception task for automated tomato harvesting. However, tomato maturity recognition in greenhouse environments is still affected by non-uniform illumination, weak light, uneven illumination and reflection, fruit overlap, cross-row plant interference, and the similarity between immature fruits and leaves. RGB images can provide rich color and texture information, but their feature representation is easily degraded under changing illumination. Near-infrared (NIR) images are less sensitive to visible light variation and can provide complementary spectral information, but their texture details are relatively weak. This study aimed to improve the robustness and adaptability of tomato maturity recognition under complex greenhouse conditions by using Red-Green-Blue and Near-Infrared (RGB-NIR) multimodal feature fusion. A dual-backbone tomato maturity recognition model named MDD-YOLOv11 was proposed on the basis of YOLOv11n-L3L4. The model was designed to enhance feature extraction, occluded target modeling, and cross-modal feature fusion. Firstly, a multi-branch convolution module (MConv) was introduced into the dual-backbone network. Multi-scale convolution branches were used to extract features with different receptive fields, so that the RGB branch could better adapt to illumination changes and the NIR branch could enhance weak feature representation. Secondly, a C3k2_DCNv4 module was embedded into the backbone and neck networks. Deformable convolution v4 was used to adaptively adjust feature sampling positions according to the geometric structure of tomato targets. This module was used to alleviate incomplete target representation caused by stems, cross-row plants, and fruit overlap, and to improve the local modeling ability for irregular and partially occluded fruits. Thirdly, a deformable cross-modal attention fusion module (DCAF) was designed at the L3 and L4 feature layers. The DCAF module used deformable alignment to reduce spatial misalignment between RGB and NIR features, and adopted window cross-attention to suppress redundant information and mine complementary features between the two modalities. In this way, more discriminative fused features were obtained for tomato maturity recognition. The public RGB-NIR tomato dataset was used in this study. The original dataset contained four illumination conditions—natural light, artificial light, weak light, and sodium yellow light—with 1,000 RGB-NIR image pairs for each condition, totaling 4,000 pairs. This study selected 2,550 image pairs from the natural light, artificial light, and weak light scenes. To reduce data leakage caused by adjacent video frames, every five consecutive RGB-NIR image pairs were grouped as one sampling unit. Using a fixed random seed, the sampling units were divided into a training set of 2,100 image pairs and a validation set of 450 image pairs. All compared models were trained and evaluated using the same dataset split and training settings. Experimental results showed that MDD-YOLOv11 achieved a Precision of 94.8%, a Recall of 94.5%, an F1 score of 94.65%, an mAP0.5 of 98.7%, and an mAP0.5:0.95 of 86.3% on the mixed illumination val set. Compared with the baseline YOLOv11n-L3L4 model, these indicators increased by 1.7, 1.6, 1.7, 1.6, and 3.4 percentage points, respectively. The number of parameters, floating-point operations (FLOPs), and model size of MDD-YOLOv11 were 4.1 M, 11.1 G, and 8.1 MB, respectively. The inference speed reached 72.7 frames/s, and the average inference latency for a single image was 0.0138 s on an NVIDIA RTX3090 GPU platform with a batch size of 1. The ablation experiments showed that MConv, C3k2_DCNv4, and DCAF all improved recognition performance. In particular, DCAF improved mAP0.5:0.95 by 2.0 percentage points when used alone, indicating that cross-modal feature alignment and fusion were effective for RGB-NIR tomato maturity recognition. Visualization results and error statistics further showed that the proposed model reduced false detections and missed detections under natural light, weak light, occlusion, and artificial light conditions. To further evaluate the adaptability of the model to different acquisition environments, a self-built HBAUT dataset containing 331 RGB-NIR image pairs after screening was used as an external test set, without participating in model training, validation, or hyperparameter tuning. MDD-YOLOv11 achieved an mAP0.5 of 95.4% and an mAP0.5:0.95 of 83.4%, which were 2.1 and 2.7 percentage points higher than those of YOLOv11n-L3L4, respectively. These results indicate that MDD-YOLOv11 can improve the robustness and adaptability of tomato maturity recognition in complex greenhouse environments, providing technical support for the visual perception stage of automated tomato harvesting.

     

/

返回文章
返回