摘要

针对空间非合作目标六自由度位姿估计问题,本文基于卷积神经网络设计一种轻量化网络LSPENet,无需手动设计特征便可实现端到端的位姿估计。本文使用深度可分离卷积、高效通道注意力机制ECA构成基本模块,兼顾了网络的复杂度和准确度。设计两个分支分别用于估计位置和姿态,位置估计使用直接回归法,姿态估计引入软分配编码。在URSO数据集的实验结果表明姿态软分配编码相比直接回归姿态能够显著减小姿态误差;本文所提网络相比于其他端到端位姿估计网络,参数量减少3倍,单幅推理时间降低13.3%,同时平均位姿估计精度提高1倍。本文实现的轻量化端到端位姿估计网络,为星载单目视觉位姿估计提供了新思路。