基于YOLOv5与RGB-D融合的交通锥桶识别与定位方法研究OA
Research on traffic cone recognition and localization method based on YOLOv5 and RGB-D fusion
针对有交通锥桶的施工与测试场景中,需实时检测交通锥桶并估算其三维位置,且在复杂光照及遮挡环境下易出现误检与漏检的问题,提出一种 YOLOv5与RGB-D深度融合的锥桶识别与定位方法.首先,采集并标注了863张涵盖多光照、多角度、多背景的锥桶图像,按8∶2比例划分为训练集与验证集,通过数据增强、超参数调优以及早停训练策略得到最优模型权重;其次,将训练好的模型部署至Intel RealSense D435相机,实现RGB图像与深度帧的同步获取与空间对齐;最后,在检测框中心区域提取深度信息,并结合相机内参进行反投影计算,完成三维坐标估计.结果表明:所提方法检测精度高(Precision≈99.8%,Recall≈99.6%,mAP@0.5≈99.5%),处理速度约26 FPS(即每帧约37 ms),平均定位误差小于5 cm,且在强光、阴影、局部遮挡等复杂条件下仍表现稳定.所提方法实现了交通锥桶二维检测与三维定位的一体化,在保证较高检测精度的同时兼顾了实时性与定位准确性,可为交通锥桶感知研究提供参考.
To meet the requirements of real-time traffic cone detection and three-dimensional localization in construction and testing scenarios involving traffic cones,and to alleviate false positives and missed detections under complex illumination and occlusion conditions,this paper proposed a traffic cone recognition and localization method based on YOLOv5 and RGB-D fusion.First,863 cone images featuring diverse lighting,viewing angles,and backgrounds were collected and annotated,and divided into training and validation sets in an 8∶2 ratio.Through data augmentation,hyperparameter tuning,and an early stopping strategy,the optimal model weights were obtained.Next,the trained model was deployed on an Intel RealSense D435 camera,achieving synchronized acquisition and spatial alignment of RGB and depth frames.Finally,depth information was extracted from the center region of the detection bounding box and combined with camera intrinsic parameters for back-projection to estimate 3D coordinates.Experimental results indicate that the proposed method has high detection accuracy(Precision≈99.8%,Recall≈99.6%,and mAP@0.5≈99.5%),with a processing speed of approximately 26 FPS(≈37 ms per frame),an average localization error of less than 5 cm.The method maintains stable performance under challenging conditions such as strong glare,shadows,and partial occlusion.This method integrates 2D detection and 3D localization of traffic cones,ensuring high detection accuracy while balancing real-time performance and localization precision.It can provide reference for traffic cone perception.
李鹏;贾正楷;杨洪玖;刘猛
天津大学电气自动化与信息工程学院,天津 300072天津大学电气自动化与信息工程学院,天津 300072天津大学电气自动化与信息工程学院,天津 300072天津大学电气自动化与信息工程学院,天津 300072
信息技术与安全科学
计算机感知交通锥桶检测三维定位深度估计实时定位
computer perceptiontraffic cone detection3D localizationdepth estimationreal-time positioning
《河北科技大学学报》 2026 (3)
229-236,8
国家自然科学基金(62403350,62373269)
评论