Laser & Optoelectronics Progress, Volume. 56, Issue 22, 221002(2019)

Pose Estimation Algorithm Based on Combined Loss Function

De Zhang, Guozhang Li, Huaiguang Wang*, and Junning Zhang
Author Affiliations
  • Department of Vehicle and Electrical Engineering, Army Engineering University Shijiazhuang Campus, Shijiazhuang, Hebei 050003, China
  • show less

    The loss function of a target pose estimation model based on a convolutional neural network (CNN) mostly uses the Euclidean distance between two points as the evaluation criterion. Although the loss function is simple in calculation and fast in operation, the training rules are not comprehensive enough and lack global understanding of the target. In this paper, a ComPoseNet model based on a combined loss function is proposed for pose estimation. The loss function in this model is based on spatial learning, and the two-point Euclidean distance, straight line, and straight line angle are used as training rules. Compared with the traditional loss function, this algorithm considers the spatial position of the target from the point, line, and angle, reducing the error between the estimated and the real poses so that the effect of the pose estimation is improved. Numerous experiments and analysis of LineMod data show that the algorithm has a higher convergence speed, greater accuracy, and smaller errors than the traditional algorithm operating for the same training times. The translation error is reduced by 7.407%, and the angle error is reduced by 6.968%.

    Tools

    Get Citation

    Copy Citation Text

    De Zhang, Guozhang Li, Huaiguang Wang, Junning Zhang. Pose Estimation Algorithm Based on Combined Loss Function[J]. Laser & Optoelectronics Progress, 2019, 56(22): 221002

    Download Citation

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

    Category: Image Processing

    Received: Mar. 4, 2019

    Accepted: May. 15, 2019

    Published Online: Nov. 2, 2019

    The Author Email: Wang Huaiguang (654959514@qq.com)

    DOI:10.3788/LOP56.221002

    Topics