An Attention-Enhanced Network for Visual Attitude Estimation

Accurate estimation of object attitude is essential for understanding motion behavior and achieving dynamic tracking. Existing image-based methods often suffer from low efficiency and limited accuracy, while the potential of deep learning has not been fully exploited in this field. To address these limitations, a lightweight deep learning method for attitude estimation is proposed and validated on spherical particles. A synthetic dataset is generated through VTK-based rendering and automatic annotation, providing large-scale training samples with known Euler angles. An improved MobileNetV1 backbone is developed by integrating Squeeze-and-Excitation blocks, a dual-scale Pyramid Pooling Module, global average pooling, and a regression-oriented multilayer perceptron, which enhances feature extraction and enables direct Euler angle prediction. Experimental results show that the proposed method achieves an average error of 0.308° on synthetic test images. Furthermore, a solid particle was fabricated through 3D printing and physical measurements were conducted, where the network combined with image preprocessing and augmentation achieved an average error of about 0.5° on real images, demonstrating a lightweight and deployment-friendly framework for practical attitude estimation. The results verify the effectiveness of the method and demonstrate its potential for accurate and computationally efficient attitude measurement in applications such as fluid dynamics, industrial inspection, and motion tracking.