ACR-Net: Attention Integrated and Cross-Spatial Feature Fused Rotation Network for Tubular Solder Joint Detection

Chenlin Zhou, Daheng Li, Peng Wang, Jia Sun, Yikun Huang, Wanyi Li

Abstract: Tubular solder joint detection is an important and challenging issue in the industry, due to the illegible objects, rarely collected datasets and requiring high-precision and real-time performance for positioning and angle estimation. In this article, we propose an Attention integrated and Cross-spatial feature fused Rotation Network (ACR-Net) for tubular solder joint detection, which consists of a new feature extraction network named ECA-CSPDarknet44, a cross-spatial feature fusion network (CFFN), and a bin-based rotation detection network (BRDN). The proposed network can efficiently detect oriented tubular solder joints with high-precision and real-time performance. ECA-CSPDarknet44 with attention mechanism was presented, which can adaptively guide the network to learn important features of tubular solder joints, significantly improving the ability of feature extraction. By integrating multiscale global features and multiscale local region features, CFFN can enhance the network's ability to express the characteristics of tubular solder joints. Meanwhile, BRDN is proposed for oriented bounding box regression through a decoupling approach, which regresses target parameters efficiently and accurately with little increasing of model complexity. Finally, we establish a tubular solder joint dataset and conduct sufficient experiments to verify the effectiveness of our method. Our proposed ACR-Net achieves 98.8% mAP with 31.3 frames per second (FPSs) on the dataset, meeting the high-precision and real-time requirements of industrial systems.