The Improved YOLOV5 Algorithm and Its Application in Small Target Detection

Abstract

According to the characteristics of YOLOV5, a method based on YOLOV5 is proposed. First, this method is used for the identification of small objects. Secondly, this paper conducts experiments on the improved YOLOV5 algorithm and conducts experiments on it, and compares the performance of the two methods. In modern detection technology, the most commonly used is the target detection technology. In recent years, small object detection has been an important research topic in industrial inspection. The small target has a small number of pixels, and its features are blurred. Larger targets have lower detection rates and higher false detection rates. In order to improve the accuracy and accuracy of target detection, this paper proposes a method based on depthwise convolution and K-means clustering, which realizes the clustering of the dimension and aspect ratio of the target image. On the basis of CCPD, the YOLOv5 algorithm is optimized and compared with the original YOLOv5 algorithm. Experiments show that the improved YOLOv5 can better detect smaller targets, so that the recall rate and average precision of small targets are significantly improved.