ZHANG Zhen1, CHEN Kexin2, CHEN Yunfei1
Abstract:
Aiming at the problems that traditional controlled knife detection methods relied too much on specific equipment and environment, and the target detection was applicable to small scope of application, with weak anti-interference ability and low precision, a YOLOv5 model with optimized clustering and CBAM was proposed for controlled knife detection was proposed. Firstly, a decision diagram could be drew and cluster centers could be obtained by calculating the local density of each point and the minimum distance between it and others with higher density. Then the core points could be extracted by calculating all points in the local class the average distance with its class center, the global search assignment strategy was used to classify the test points. Finally, the statisticallearning strategy was employed to allocate the remaining points, these unprocessed points were used as noise points, and they were classified into the class of its nearest neighbor. The improved density peak clustering algorithm was used to analyze the bounding box of the controlled knife, optimize the size of the priori box, and improve the matching degree between the priori box and the size of target object, so as to solve the problem that the clustering effect of the K-means clustering algorithm in the YOLOv5 model such as its unstability and the low convergence rate of large-scale data. Moreover,The C3 module in the Backbone network was combined with the CBAM attention mechanism, named CBAMC3 module, which could improve the model′s ability to extract target features, solve the problem that the YOLOv5 algorithm was not effective for small target detection, and improve the model accuracy. The experimental results showed that the values of P, R, mAP @ 0. 5, mAP @ 0. 5 ∶ 0. 95 of the improved model YOLOv5-Plus on the customed data set were 98. 14%, 95. 80%, 97. 56%, and 76. 68%, respectived, which is 1. 64%, 1. 59%, 1. 51%, and 3. 26% higher than that of YOLOv5 before improvement, and also verified that the proposed model could effectively improve the detection performance of controlled knife in public areas.