<p>6-Dof pose estimation of texture-less and highly reflective workpieces is an important application of computer vision, especially in the field of robot vision application. In this study, we propose a Swin transformer-based network named Swin-AFF, a method based on a single RGB-D image which is suitable for highly accurate 6D pose estimation of workpiece with texture-less or highly reflective from occluded scenes. The geometric and texture features of the workpiece are effectively extracted from the image feature coding and decoding network based on Swin transformer by using RGB images combined with the normal vector angle images generated by the depth channel, while the geometric features of the point cloud of the target are effectively extracted by using Rand-LA network. Meanwhile, a bidirectional feature fusion module with the function of suppressing abnormal noise of images based on adaptive fusion strategy is constructed. Finally, in order to improve accuracy and efficiency, the 3D-3D algorithm is used to calculate the 6D pose of the object based on the predicted target mask and 3D key points. In addition, a new dataset (HW6D) containing highly reflective symmetrical metal workpieces and texture-less plastic workpieces with different shapes and structures is constructed to verify the proposed method. Using ADD(S) evaluation metric, the experiments on the LineMOD, MP6D and HW6D datasets show that our method outperforms state-of-the-art methods for 6D pose estimation. Ablation research validates the design of Swin-AFF. HW6D dataset and Swin-AFF code are available at: <a href="https://doi.org/10.6084/m9.figshare.26411209.v2">https://doi.org/10.6084/m9.figshare.26411209.v2</a>.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Swin-AFF: an improved accuracy 6D pose estimation network for high reflection and texture-less workpieces based on Swin transformer

  • Zhentao Li,
  • Zhiyang Guo,
  • Yun Feng,
  • Guanjun Zhang,
  • Panfeng Wang,
  • Wenlei Wu

摘要

6-Dof pose estimation of texture-less and highly reflective workpieces is an important application of computer vision, especially in the field of robot vision application. In this study, we propose a Swin transformer-based network named Swin-AFF, a method based on a single RGB-D image which is suitable for highly accurate 6D pose estimation of workpiece with texture-less or highly reflective from occluded scenes. The geometric and texture features of the workpiece are effectively extracted from the image feature coding and decoding network based on Swin transformer by using RGB images combined with the normal vector angle images generated by the depth channel, while the geometric features of the point cloud of the target are effectively extracted by using Rand-LA network. Meanwhile, a bidirectional feature fusion module with the function of suppressing abnormal noise of images based on adaptive fusion strategy is constructed. Finally, in order to improve accuracy and efficiency, the 3D-3D algorithm is used to calculate the 6D pose of the object based on the predicted target mask and 3D key points. In addition, a new dataset (HW6D) containing highly reflective symmetrical metal workpieces and texture-less plastic workpieces with different shapes and structures is constructed to verify the proposed method. Using ADD(S) evaluation metric, the experiments on the LineMOD, MP6D and HW6D datasets show that our method outperforms state-of-the-art methods for 6D pose estimation. Ablation research validates the design of Swin-AFF. HW6D dataset and Swin-AFF code are available at: https://doi.org/10.6084/m9.figshare.26411209.v2.