Aiming at the problems of large recognition error and low detection accuracy of small targets in target detection, this paper proposes a multi-target detection algorithm based on deep learning. This method is improved on the basis of YOLOv3. First, the Darknet network is modified, and a set of residual blocks is added at the end of the network to obtain feature maps of 4 scales. Second, replace IOU with DIOU to increase network robustness. Finally, the K-means++ algorithm is used to generate 12 anchor boxes on the training set. The results show that when the input is 576x576, based on the Nvidia 980Ti graphics card, the average accuracy of this algorithm on the VOC 2007 data set reaches 80.1% and the speed reaches 52.64fps, which improves the detection accuracy of multi-target objects while ensuring real-time performance.
|