YOLO-DTO: Automotive Door Panel Fastener Detection Algorithm Based on Deep Learning

Abstract

The common detection of fasteners of automobile door panels is based on the method of template matching, which has the problems of low detection accuracy and poor real-time performance under the influence of different lighting and different placement positions. To improve the detection speed and accuracy of fasteners in complex scenes, a small object detection algorithm, YOLO-DTO (Detect Tiny Object), was proposed based on the YOLOv8 algorithm. Firstly, considering that the algorithm uses strided convolution to compress the input image prematurely, resulting in the loss of fine-grained information in the early stage of the image, which makes it difficult to recover the complete detail information in the subsequent feature fusion process, this paper modifies the convolution module in the early stage of the algorithm and introduces the SPD (SPace-to-Depth) module to reconstruct the early stage of the original algorithm. Secondly, a selective attention module is embedded in the Neck output position of the algorithm to enhance the algorithm's ability to pay attention to the context information of fasteners. Finally, to optimize the regression efficiency of the bounding box, the MPDIOU loss function replaced the CIOU loss function. Experimental results show that the average detection accuracy of the YOLO-DTO algorithm is 98.8%, which is 9.1% and 1.7% higher than that of the template matching method and YOLOv8 algorithm, respectively, which meets the detection standards of factory production lines and has the practical value.