Welcome to Journal of Graphics share: 

Journal of Graphics ›› 2021, Vol. 42 ›› Issue (2): 198-205.DOI: 10.11996/JG.j.2095-302X.2021020198

• Image Processing and Computer Vision • Previous Articles     Next Articles

Real-time gun detection method based on compressed YOLOV3 

  

  1. 1. School of Information Engineering, Engineering University of PAP, Xi’an Shaanxi 710086, China;  2. Counter-Terrorism Command Information Engineering Research Team, Engineering University of PAP, Xi’an Shaanxi 710086, China;  3. Graduate Team, Engineering University of PAP, Xi’an Shaanxi 710086, China
  • Online:2021-04-30 Published:2021-04-30
  • Supported by:
    National Natural Science Foundation of China (U1603261); Natural Science Foundation of Xinjiang Uygur Autonomous Region (2016D01A080) 

Abstract: 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. 

Key words: gun detection, YOLOV3, model compression, K-means ++, real-time detection 

CLC Number: