Chinese Journal of Liquid Crystals and Displays, Volume. 37, Issue 7, 913(2022)

Object 6D pose estimation algorithm based on improved heatmap loss function

Lin LIN1,2, Yan-jie WANG1、*, and Hai-chao SUN1
Author Affiliations
  • 1Changchun Institute of Optics, Fine Mechanics and Physics, Chinese Academy of Sciences, Changchun 130033, China
  • 2University of Chinese Academy of Sciences, Beijing 100049, China
  • show less

    In view of the problem of low precision and slow training of heatmap regression network trained by mean square error (MSE) loss function used in traditional heatmap regression, the loss function Heatmap Wing Loss (HWing Loss) for heatmap regression is proposed in this thesis. In terms of different pixel values, the loss function has different loss function values, and the loss function gradient of foreground pixels is larger, which can make the network focus more on the foreground pixels and make the heatmap regression more accurate and faster. In line with the distribution characteristics of the heatmap, the keypoint inference method based on the Gaussian distribution is adopted in this thesis to reduce the quantization error when the heatmap infers the keypoints. By taking the two points as the basis, it constructs a new monocular pose estimation algorithm based on keypoint positioning. According to the experiments, in contrast with the algorithm using MSE Loss, the pose estimation algorithm using HWing Loss has a higher ADD(-S) accuracy rate, which reaches 88.8% on the LINEMOD dataset. Meanwhile, the performance is better than other recent pose estimation algorithms based on deep learning. The algorithm in this thesis can run at the fastest speed of 25 fps on RTX3080 GPU, in which the high speed and performance can be both embodied.

    Tools

    Get Citation

    Copy Citation Text

    Lin LIN, Yan-jie WANG, Hai-chao SUN. Object 6D pose estimation algorithm based on improved heatmap loss function[J]. Chinese Journal of Liquid Crystals and Displays, 2022, 37(7): 913

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Dec. 3, 2021

    Accepted: --

    Published Online: Jul. 7, 2022

    The Author Email: Yan-jie WANG (wangyj@ciomp.ac.cn)

    DOI:10.37188/CJLCD.2021-0317

    Topics