不确定性感知的RGB图像6D物体位姿估计方法OA
Uncertainty-aware monocular RGB 6D object pose estimation method
针对复杂背景与弱纹理条件下6D物体位姿估计精度受限的问题,本文提出了一种不确定性感知的位姿估计框架.首先,在特征提取阶段于编码器-解码器结构中引入多尺度感受野增强模块,通过融合不同卷积尺度与空洞率的上下文信息,提升网络对弱纹理目标的多尺度表征能力.其次,设计像素级置信度感知投票机制,对关键点密集单位向量场预测结果进行置信度建模,并在关键点投票过程中对像素贡献进行自适应加权,以提高关键点定位的鲁棒性.最后,将关键点定位结果的不确定性显式引入姿态求解过程,构建不确定性感知的PnP位姿估计框架,以减弱噪声关键点对姿态估计的影响.在 LineMOD,Occlusion LineMOD及 YCB-Video数据集上的实验结果表明,与 BB8,DenseFusion,PoseCNN等多种方法相比,所提方法在位姿估计精度方面取得显著提升,平均距离(ADD)指标平均提高约7%,并在复杂背景与弱纹理场景下表现出良好的鲁棒性.实验结果充分验证了该方法的有效性与复杂场景适应能力.
To address the limited accuracy of 6D object pose estimation under complex backgrounds and weak-texture conditions,an uncertainty-aware 6D object pose estimation framework was proposed.First-ly,a multi-scale receptive field enhancement module was introduced into the encoder-decoder architec-ture at the feature extraction stage.By integrating contextual information with different convolutional scales and dilation rates,the network was able to effectively improve the multi-scale feature representation capability for weak-texture objects.Secondly,a pixel-wise confidence-aware voting mechanism was de-signed to model the reliability of the predicted dense keypoint unit vector fields,and the contribution of each pixel was adaptively weighted during the voting process,thereby enhancing the robustness and accura-cy of keypoint localization.Finally,the uncertainty of keypoint localization was explicitly incorporated into the pose estimation stage by constructing an uncertainty-aware PnP framework,which effectively reduced the influence of noisy keypoints on pose estimation results.Experimental results on the LineMOD,Occlu-sion LineMOD,and YCB-Video datasets demonstrate that,in comparison with several representative methods such as BB8,DenseFusion,and PoseCNN,the proposed method achieves significant improve-ments in pose estimation accuracy,with an average increase of approximately 7%in the Average Distance(ADD)metric,and exhibits strong robustness under complex background and weak-texture scenarios.These results demonstrate the effectiveness,robustness,and adaptability of the proposed method under complex background and weak-texture conditions.
李露帆;党建武;雍玖
兰州交通大学 电子与信息工程学院,甘肃兰州 730070兰州交通大学 电子与信息工程学院,甘肃兰州 730070||兰州交通大学 轨道交通信息与控制国家级虚拟仿真实验教学中心,甘肃兰州 730070||兰州大成科技股份有限公司,甘肃兰州 730070兰州交通大学 电子与信息工程学院,甘肃兰州 730070||兰州交通大学 轨道交通信息与控制国家级虚拟仿真实验教学中心,甘肃兰州 730070||兰州大成科技股份有限公司,甘肃兰州 730070
信息技术与安全科学
物体位姿估计关键点定位像素级置信度投票不确定性感知PnP位姿求解
6D object pose estimationkeypoint localizationpixel-wise confidence-aware votinguncer-tainty-aware modelingPnP pose estimation
《光学精密工程》 2026 (16)
2595-2611,17
国家自然科学基金项目(No.62367005)铁路青年人才托举工程项目(第七届)中国高等教育学会高校数字思政精品项目(No.GJXHSZSZZY047)兰州交通大学"科研反哺教学典型案例"教学改革研究项目(No.JGZ202503)
评论