计算机科学 ›› 2026, Vol. 53 ›› Issue (6A): 250700173-8.doi: 10.11896/jsjkx.250700173
董烨1, 连心悦1, 王雨阳1, 欧新宇1,2
DONG Ye1, LIAN Xinyue1, WANG Yuyang1, OU Xinyu1,2
摘要: 番茄自动化采摘在现代农业生产中至关重要,能够显著提升效率,降低成本,并保障果实品质。然而,现有基于单一模态的自动采摘技术在果园环境中仍面临诸多挑战:复杂光照条件与枝叶遮挡容易导致番茄目标出现误检和漏检;同时,番茄作为小目标,在传统RGB图像中的特征易被背景噪声淹没;此外,单一传感器数据难以全面捕捉目标特征,限制了模型的鲁棒性与适应性。针对上述问题,提出一种基于多模态特征融合的RTDMF模型(Robust Tomato Detection with Multi-modal Fusion),通过整合RGB与IR图像的互补特性,增强番茄检测的精准性与稳定性。该模型在YOLOv5基础上进行优化,针对小目标检测引入轻量化深度可分离卷积与自适应锚框策略,以提升对细微特征的捕捉能力;采用双分支架构分别处理RGB和IR图像,并借助自注意力机制实现多模态特征的高效融合,充分结合色彩、纹理与温度信息;此外,引入Mosaic数据增强与动态学习率策略,进一步提升模型的泛化性能与收敛效率。实验结果表明,在包含多种光照与遮挡条件的“Multimodal image dataset of tomato fruits with different maturity”数据集上,RTDMF模型将使平均精度(mAP)提升9.7%,召回率提高0.6%,漏检率降低2.3%,误检率降低3.1%。可视化结果进一步验证了该模型在低对比度、遮挡等复杂场景下的强适应性。
中图分类号:
| [1] RAM P P V S,YASWANTH K V S,KAMEPALLI S S B S,et al.Deep learning model YOLOv5 for red chilies detection from chilly crop images[C]//Proc of the 8th IEEE Int Conf on Technology Fusion(I2CT).IEEE,Piscataway,NJ,2023. [2] ZHANG Y,RAO Y,CHEN W J,et al.Multimodal image dataset of tomato fruits with different maturity[DB/OL].Science Data Bank.Elsevier,2023. [3] LI S Q,TANG L,LIU K Y,et al.A fast and adaptive objecttracking method[J].Journal of Computer Research and Deve-lopment,2012,49(2):383-391. [4] XU S F,CHEN X,LI H W,et al.Airborne small target detection method based on multimodal and adaptive feature fusion[J].IEEE Transactions on Aerospace and Electronic Systems,2024,62(2):1-15. [5] ZHAO J M,SHI Z L,YU C,et al.Infrared small target detection based on adjustable sensitivity strategy and multi-scale fusion[J].Infrared Physics & Technology,2024,2407(20090):1-14. [6] BADRINARAYANAN V,KENDALL A,CIPOLLA R.Seg-Net:A deep convolutional encoder-decoder architecture for image segmentation[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2017,39(12):2481-2495. [7] NIU S Q,XU X L,LIANG A,et al.Research on a lightweight method for maize seed quality detection based on improved YOLOv8[J].IEEE Access,2024,12:32927-32937. [8] WANG C Y,BOCHKOVSKIY A,LIAO H Y.YOLOv7:Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors[C]//Proc of the 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR 2021).IEEE,Vancouver,BC,Canada,2023. [9] SHANG Y Y,ZHANG Q R,SONG H B.Application of deep learning using YOLOv5s to apple flower detection in natural scenes[J].Transactions of the Chinese Society of Agricultural Engineering(Transactions of the CSAE),2022,38(9):222-229. [10] OMER S M,GHAFOOR K,ASKAR S.Lightweight improvedthe YOLOv5 model for cucumber leaf disease and pest detection based on deep learning.Computers and Electronics in Agriculture[J].Signal,Image and Video Processing,2024,18:1329-1342. [11] WANG F Z,WANG P,ZHANG X,et al.An overview of parametric modeling and methods for radar target detection with limited data[J].IEEE Access,2021,9:60459-60469. [12] LIU L B,CHEN J Q,WU H F,et al.Cross-modal collaborative representation learning and a large-scale RGBT benchmark for crowd counting[C]//2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR).IEEE,2021:479-488. [13] XIE J,NIE J,DING B A,et al.Cross-Modal Local Calibration and Global Context Modeling Network for RGB-Infrared Remote-Sensing Object Detection[J]. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing,2023,16:8933-8942. [14] MEES O,EITEL A,BURGARD W.Choosing smartly:Adaptive multimodal fusion for object detection in changing environments[C]//Proc of the IEEE/RSJ International Conference on Intelligent Robots and Systems(IROS).Daejeon,Korea(South):IEEE,2016:775-9048. [15] TANG H,LI Z C,ZHANG D,et al.Divide-and-Conquer:Confluent Triple-Flow Network for RGB-T Salient Object Detection[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2025,47(3):1958-1974. [16] TIAN H K,SONG K C,TONG L,et al.Robot Unknown Objects Instance Segmentation Based on Collaborative Weight Assignment RGB-Depth Fusion[J].IEEE/ASME Transactions on Mechatronics,2024,29(3):2032-2043. [17] WANG M C,WANG H,LI Y C,et al.MSAFusion:Object Detection Based on Multisensor Adaptive Fusion Under BEV[J]. IEEE Transactions on Instrumentation and Measurement,2025,74:9509212. [18] WANG L C,LI C X.Research on multimodal fusion object detection algorithm of image design and computational vision[C]// International Proc of 2024 5th International Conference on Computer Vision,Image and Deep Learning(CVIDL).2024:19-21. [19] CHEN J,REN H Z,YANG H T,et al.M2FNet:Multi-modal fusion network for object detection from visible and thermal infrared images[J].International Journal of Applied Earth Observation and Geoinformation,2024,130(103918):1-16. [20] YUN S D,HAN D Y,OH S J,et al.CutMix:Regularizationstrategy to train strong classifiers with localizable features[C]//Proc of IEEE/CVF CVPR.2019:10.27-11.02:612. [21] ZHONG Z,LIANG Z,KANG G L,et al,Random Erasing data augmentation[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2020:13001-13008. [22] DAI H C,WEI X Y,XU Y X,et al.Multimodal fusion detection method for tomato fruits based on phase and hyperspectral imaging[J].Acta Photonica Sinica,2024,53(1):1-10. [23] LI R J,SONG T,GAO J,et al.Tomato diseased leaf detection model in natural environment based on improved YOLOv5[J].Jiangsu Journal of Agricultural Sciences,2024,40(6):1028-1037. |
|
||