A Lightweight Detection Method of Smartphone Assembly Parts

Bo Zhang,Wenbai Chen,Xiaohao Wang,Chunjiang Zhao

DOI: https://doi.org/10.1007/978-3-031-20497-5_27

2022-01-01

Abstract:The object detection algorithm for smartphone assembly parts in the 3C (Computer, Communication, Consumer Electronics) scene occupies many system computing resources, and the flexible targets and small-scale heterogeneous components in the background lead to problems such as low detection accuracy. Based on the YOLOv5n network, we propose a lightweight and high-precision network-YOLOv5nGTA. First, we replace the ordinary convolution module with the Ghost convolution module in the backbone network and neck, thereby significantly reducing the parameter scale in the network. Second, we add a Transformer module at the end of the backbone network to enhance the feature expression ability of the network and further improve the modeling and representation ability of the backbone network. Then, We use the meta-ACON (Activate or not) activation function to dynamically learn the activation function's linearity and solve the neuron necrosis problem by controlling the nonlinearity of each layer of the network. Experimental results show that our method outperforms other excellent detection algorithms in inference speed and model size evaluation indicators. We compress model parameters to 63% of YOLOv5n and model size to 75% of YOLOv5n at a speed of 24.5 ms per image on GPU.