基于神经网络特征线提取的飞机位姿识别方法研究

Research of aircraft pose estimation based on neural network feature line extraction

  • 摘要: 为实现复杂情况下的飞机位姿识别,提出了基于神经网络特征线提取的位姿识别新方法。该方法利用3D模型进行图像渲染,通过添加背景形成数据集,为提高算法鲁棒性进行了数据集增强。特征线提取模型采用卷积神经网络提取目标深度特征,利用热力图获取飞机特征线。结合飞机特征线、飞机3D模型以及n线透视方法解算目标位姿。该方法建立的飞机特征线提取模型,在复杂背景下准确率约为91%。叠加了各类噪声后,准确率为84%。飞机位姿通过EPnL算法与非线性优化进行求解。在目标背景复杂的情况下,实验得到的平均预测角度误差约为0.57°,平均预测平移误差约为0.47%。图像叠加各类噪声后,得到的平均预测角度误差约为2.11°,平均预测平移误差约为0.93%。提出的飞机位姿识别方法在复杂背景、各类噪声影响下可以较精准地预测飞机位姿,应用场景更加广泛。

     

    Abstract: To estimate the aircraft pose in complex situation, this paper proposes a new method of aircraft pose estimation based on neural network line extraction. This method uses 3D model to render images, and forms dataset through adding backgrounds. The dataset is enhanced to make the algorithm robust. The line extraction model uses convolutional neural network to extract deep features, and uses heatmap to obtain aircraft feature lines. The target pose is solved by combining the aircraft feature line, the aircraft 3D model and the perspective-n-line method. The accuracy of the line extraction model is 91% in complex background. The accuracy is 84% after addingall sorts of noises. The aircraft pose is solved by using EPnL algorithm and nonlinear optimization. The average angle error is about 0.57°, and the average translation error is about 0.47% when the target is in a complex background. After addingall sorts of noises to the image, the average angle error is about 2.11°, and the average translation error is about 0.93%. The aircraft pose estimation method proposed in this article can accurately predict the aircraft pose under complex backgrounds and various types of noise, and its application scenarios are more extensive.

     

/

返回文章
返回