Improvement of lithium battery corner detection accuracy based on image restoration method

Abstract

Target detection technology has been widely used in the automatic production of lithium batteries. However, motion blur will lead to the reduction of the angular position detection accuracy of lithium batteries. To solve this problem, an improved fuzzy recovery model for angular position of lithium battery is proposed in this paper. Firstly, the improved lightweight neural network RepVGG was used as the main module of the backbone network, so that the network could improve the performance of network feature extraction while reducing the number of calculation parameters and improving the reasoning speed of fuzzy restoration. Secondly, we optimize the multi-Dconv head transposed attention (MDTA) module and reference it to the generator, which reduces the complexity of the model and strengthens the network’s attention to details and textures, and improves the visual effect of the restored image. Finally, we design a lightweight globally connectable residual network called SAC Block and use it to to improve the discriminator, which enhances the global receptive field of the model and improves the structural similarity between the restored image and the original image. In order to verify the effectiveness of the method, we verify it on the self-built dataset and GoPro dataset. The experiments show that our proposed lightweight model improves the peak signal-to-noise ratio (PSNR) index by 9.2% and 8.6% respectively compared with the original model. The visual effect of the restored image is better than that of other current similar algorithms, and it is confirmed that our model can better improve the accuracy of lithium battery angular position detection.