Fast wafer detection algorithm based on YOLOv5s

In order to solve the problem that traditional wafer localization methods are easily affected by light, noise, etc., and that deep learning target detection methods are slow for wafer detection, we constructed the wafer fast detection algorithm YOLOv5s-fast. firstly, in the feature extraction part, we used the FasterNet network for feature extraction, which accelerated the algorithm running speed; secondly, in order to improve the algorithm detection performance, we added BRA bi-directional routing attention (Bilevel Routing Attention) to feature fusion part of the network to strengthen the feature fusion capability. Secondly, in order to improve the detection performance of the algorithm, BRA Bilevel Routing Attention is added in the feature fusion part of the network to strengthen the feature fusion capability. Finally, the MPDIOU loss function is used to improve the model detection performance. The average detection accuracy of the algorithm is 99.50%, the number of parameters is 1064176, and the detection speed is 2.8GFLOPs, which improves both the detection speed and the detection accuracy, and at the same time realizes the algorithm lightweight.