基于小波特征增强与自训练的域迁移航天器目标6D位姿估计

Domain-migration Spacecraft 6D Pose Estimation via Wavelet-based Feature Enhancement and Self-training

  • 摘要: 非合作航天器目标如空间自主交会对接的目标航天器、执行任务失效的航天器以及执行在轨维修任务的航天器,需要快速准确地解算6D相对位姿才能够进行后续服务航天器的制导与控制。以前的方法通常采用CNN(卷积神经网络)以及Transformer架构进行远程的特征交互,但下采样等操作会忽略边缘等高频特征而造成特征失真。此外,研究者设计多头损失函数或引入对抗学习进行位姿估计的迁移学习,但任务复杂且费时。相比而言,本文提出一种基于小波特征增强的航天器跨域位姿估计方法。首先,将提取的多尺度特征分解为高频带与低频带,采用有效的Transformer架构在频域层面执行空间与上下文的注意操作,增强特征表达能力;其次,对处理后的特征进行小波分解,增强低分辨率图的高频特征;最后,将源域训练好的模型直接用于目标域的位姿预测,并根据所获得的伪标签进行迭代自训练。在SwissCube、SPEED、SPEED+ 位姿估计数据集上的综合实验表明,所提出方法能够有效提高位姿估计的精度,并且推理速度具有竞争力。

     

    Abstract: Non-cooperative spacecraft targets, such as spacecrafts engaged in autonomous rendezvous and docking, spacecrafts that have failed during missions, and spacecrafts performing on-orbit maintenance tasks, require rapid and accurate 6D relative pose estimation for subsequent guidance and control of service spacecraft. Previous methods often utilize CNNs (convolutional neural networks) and transformers for long-range feature interactions. However, the operations like downsampling tend to neglect high-frequency features such as edges, resulting in feature distortion. Additionally, researchers have designed multi-head loss functions or incorporated adversarial learning to perform transfer learning for pose estimation, but the tasks are complicated and time-consuming. In comparison, this paper proposes a wavelet feature enhancement approach to achieve accurate pose estimation for cross-domain spacecraft. Firstly, the extracted multi-scale features are decomposed into high-frequency and low-frequency bands. An effective transformer is used to perform spatial and contextual attention operations in the frequency domain, thus improving feature representation. Secondly, wavelet decomposition is performed on the processed features to enhance the high-frequency features of low-resolution images. Finally, the model trained in the source domain is directly applied to pose prediction in the target domain, and iterative self-training is conducted based on the obtained pseudo-labels. Comprehensive experiments on the SwissCube, SPEED, and SPEED+ pose estimation datasets demonstrate that the proposed method can effectively improve pose estimation accuracy and achieve competitive inference speed.

     

/

返回文章
返回