SESPnet: a lightweight network with attention mechanism for spacecraft pose estimation

被引:0
作者
Chen C. [1 ]
Jing Z. [1 ]
Pan H. [1 ]
Dun X. [1 ]
Huang J. [1 ]
Wu H. [2 ]
Cao S. [2 ]
机构
[1] School of Aeronautics and Astronautics, Shanghai Jiao Tong University, Shanghai
[2] Shanghai Aerospace Control Technology, Shanghai
基金
中国国家自然科学基金;
关键词
Attention mechanism; Convolutional neural network; Embedded application; Spacecraft pose estimation;
D O I
10.1007/s42401-023-00259-w
中图分类号
学科分类号
摘要
Spacecraft pose estimation plays an important role in an increasing number of on-orbit services: rendezvous and docking, formation flights, debris removal, and so on. Current solutions achieve excellent performance at the cost of a huge number of model parameters and are not applicable in space environments where computational resources are limited. In this paper, we present the Squeeze-and-Excitation based Spacecraft Pose Network (SESPNet). Our primary objective is to make a trade-off between minimizing model parameters and preserving performance to be more applicable to edge computing in space environments. Our contributions are primarily manifested in three aspects: first, we adapt the lightweight PeleeNet as the backbone network; second, we incorporate the SE attention mechanism to bolster the network’s feature extraction capabilities; third, we adopt the Smooth L1 loss function for position regression, which significantly enhances the accuracy of position estimation. © Shanghai Jiao Tong University 2023.
引用
收藏
页码:1 / 10
页数:9
相关论文
共 26 条
[1]  
Nanjangud A., Blacker P.C., Bandyopadhyay S., Gao Y., Robotics and ai-enabled on-orbit operations with future generation of small satellites, Proc IEEE, 106, 3, pp. 429-439, (2018)
[2]  
Taylor B., Aglietti G., Fellowes S., Ainley S., Salmon T., Retat I., Burgess C., Hall A., Chabot T., Kanan K., Et al., Remove debris mission, from concept to orbit, Smallsat 2018-32Nd Annual AIAA/USU Conference on Small Satellites, pp. 1-10, (2018)
[3]  
Hoque S., Arafat M.Y., Xu S., Maiti A., Wei Y., A comprehensive review on 3d object detection and 6d pose estimation with deep learning, IEEE Access, 9, pp. 143746-143770, (2021)
[4]  
Kendall A., Grimes M., Cipolla R., Posenet: A convolutional network for real-time 6-dof camera relocalization, Proceedings of the IEEE International Conference on Computer Vision, pp. 2938-2946, (2015)
[5]  
Szegedy C., Liu W., Jia Y., Sermanet P., Reed S., Anguelov D., Erhan D., Vanhoucke V., Rabinovich A., Going deeper with convolutions, Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 1-9, (2015)
[6]  
Hesch J.A., Roumeliotis S.I., A direct least-squares (Dls) method for pnp, 2011 International Conference on Computer Vision, pp. 383-390, (2011)
[7]  
Park T.H., Sharma S., D'Amico S., Towards Robust Learning-Based Pose Estimation of Noncooperative Spacecraft, (2019)
[8]  
Kisantal M., Sharma S., Park T.H., Izzo D., Martens M., D'Amico S., Satellite pose estimation challenge: Dataset, competition design, and results, IEEE Trans Aerosp Electron Syst, 56, 5, pp. 4083-4098, (2020)
[9]  
European Space Agency: Spacecraft Pose Estimation Challenge 2021, (2021)
[10]  
Chen B., Cao J., Parra A., Chin T.-J., Satellite pose estimation with deep landmark regression and nonlinear pose refinement, Proceedings of the IEEE/CVF International Conference on Computer Vision Workshops, (2019)