基于模型压缩的YOLOV3实时枪支识别方法

Real-time gun detection method based on compressed YOLOV3

以枪支为代表的危险目标检测一直是安防领域的重要研究之一。针对当前人工通过监控视频检查枪支等危险物效率低且准确率易受检查人员工作时长影响的问题,提出了利用剪枝方法对YOLOV3模型做压缩的实时枪支检测方法。采用K-means++算法对图像样本进行锚定框Anchor大小聚类,以提高模型精度。利用“通道+层”剪枝方法将训练后的模型进行压缩,通过模型修正恢复压缩前的精度。实验结果表明,该方法在保持较高精度的情况下,不仅降低了模型对内存资源的占用,且进一步减少计算量,大大提高了模型推理速度。与YOLOV3方法相比,该方法在jetson nano平台上对模型参数的缩减比例达到1/52,推理速度提高了6倍,而精确度几乎保持不变,从而达到对枪支危险物检测的实时性和高精度要求。

The detection of such dangerous targets as guns has always been one of the important research subjects in the field of security.Manual inspection of guns and other dangerous objects through surveillance video is inefficient,the accuracy of which can be easily affected by unfavorable conditions of inspectors due to continuous working.Therefore,a real-time gun detection method using the pruning method to compress the YOLOV3 model was proposed.In order to improve model accuracy,the K-means++algorithm was adopted to cluster Anchor size of data set.Then the pruning method of“channel+layer”was employed to compress the trained model.Finally,by means of retraining,the accuracy was restored to that before compression.The experimental results show that while still maintaining high accuracy,this method can reduces not only the model’s occupation of memory resources,but also the computing load,greatly improving the inference speed of the model.Compared with the YOLOV3 method,this method reduces the model parameters by up to 1/52 on the jetson nano platform and increases the reasoning speed by 6 times,with the accuracy almost unchanged,thus meeting the requirements of real-time and high-precision detection of such dangerous objects as guns.