Optimization of target detection model based on Yolov5 network with small samples

Abstract

As an important part of automotive shock absorber, the columnar parts in automotive shock absorber will inevitably have machining defects during the process, which will not only degrade the performance of the parts, but also degrade or even fail the performance of the final shock absorber after assembly. Yolov5, as a target detection algorithm, has received much attention due to its high accuracy and fast operation speed. However, the algorithm faces some challenges when applied in a practical industrial environment. In this paper, improvement measures are proposed to address the limitations of sample collection and the high speed of pipeline recognition in industrial environments. The network model is optimized and designed. Firstly, the ASPP module is replaced by the SPP module thus improving the viewability throughout the process providing recognition accuracy. Secondly, the Conv and C3 layers of Yolov5s are replaced by Transformer to obtain higher recognition accuracy. By improving and optimizing the above methods, we can better cope with the improvement of detection accuracy under small sample conditions. Experiments show that the method can significantly improve the detection accuracy and operation speed of Yolov5s under the hardware condition of lower computing power, which is more suitable for industrial scenario application scenarios.