首页|期刊导航|中国瓜菜|基于YOLO-TM的番茄采摘机器人感知算法

基于YOLO-TM的番茄采摘机器人感知算法OA

Sensing algorithm for tomato harvesting robot based on YOLO-TM

中文摘要英文摘要

为解决番茄采摘机器人作业中由类间和类内遮挡、光照变化复杂和尺度差异大等因素引起的检测精度不足的问题,提出一种基于YOLO-TM(YOLO-transformer for tomatoes maturity detection)的番茄成熟度检测模型.首先,在骨干网络引入多头自注意力机制(multi-head self-attention,MHSA)以增强全局特征提取并抑制背景干扰;然后,构建双向自适应特征金字塔网络(bidirectional adaptive feature pyramid network,BAFPN)以提升多尺度特征融合能力;此外,设计Lbox回归损失函数优化对小尺度番茄目标的定位精度.试验结果表明,YOLO-TM在自制番茄成熟度检测数据集中平均精度均值(mAP)达到 95.3%,推理速度达到 94.6 帧·s-1.与基线模型YOLOv11 相比,其mAP提升 4.2 个百分点,且实地采摘成功率达到 94.0%.与Faster R-CNN及其他YOLO系列主流模型相比,YOLO-TM在保持高实时性的同时明显提升了检测精度,展现出在复杂温室环境下的优越适应性,为番茄的自动化精准采摘及智能农业装备的视觉感知系统研发提供了坚实的理论依据与技术支持.

To address the issue of insufficient detection accuracy in tomato harvesting robot operations caused by factors such as inter-class and intra-class occlusion,complex lighting variations,and larger scale differences,this paper proposes a tomato maturity detection model based on YOLO-TM(YOLO-transformer for tomatoes maturity detection).First,a multi-head self-attention(MHSA)mechanism is introduced into the backbone network to enhance global feature extrac-tion and suppress background interference.Then,a bidirectional adaptive feature pyramid network(BAFPN)is construct-ed to improve multi-scale feature fusion capability.Furthermore,a Lbox regression loss function is designed to optimize the localization accuracy of small-scale tomato targets.The experimental results show that YOLO-TM achieves a mean aver-age precision(mAP)of 95.3%and an inference speed of 94.6 frames per second(FPS)on a self-collected tomato maturity detection dataset.Compared with the baseline model YOLOv11,YOLO-TM improves mAP by 4.2 percentage points,and achieves a field picking success rate of 94.0%.Compared with Faster R-CNN and other mainstream YOLO series models,YOLO-TM significantly improves detection accuracy while maintaining high real-time performance,demonstrating supe-rior adaptability in complex greenhouse environments.This study provides a strong theoretical basis and technical support for the automated precise picking of tomato and the development of visual perception systems for intelligent agricultural equipment.

张文娟;胡海州;杨聪敏

河南科技职业大学机电工程学院 河南 周口 466000郑州大学电气与信息工程学院 郑州 450001河南农业大学信息与管理科学学院 郑州 450046

农业科技

番茄成熟度YOLO-TM损失函数小尺度目标

TomatoRipenessYOLO-TMLoss functionSmall-scale target

《中国瓜菜》 2026 (4)

60-71,12

国家自然科学基金(61903341)

10.16861/j.cnki.zggc.2025.0567

评论