Development and Optimization of Deep Learning Models for Weapon Detection in Surveillance VideosShow others and affiliations
2022 (English)In: Applied Sciences, E-ISSN 2076-3417, Vol. 12, no 12, article id 5772Article in journal (Refereed) Published
Abstract [en]
Featured Application This work has applied computer vision and deep learning technology to develop a real-time weapon detector system and tested it on different computing devices for large-scale deployment. Weapon detection in CCTV camera surveillance videos is a challenging task and its importance is increasing because of the availability and easy access of weapons in the market. This becomes a big problem when weapons go into the wrong hands and are often misused. Advances in computer vision and object detection are enabling us to detect weapons in live videos without human intervention and, in turn, intelligent decisions can be made to protect people from dangerous situations. In this article, we have developed and presented an improved real-time weapon detection system that shows a higher mean average precision (mAP) score and better inference time performance compared to the previously proposed approaches in the literature. Using a custom weapons dataset, we implemented a state-of-the-art Scaled-YOLOv4 model that resulted in a 92.1 mAP score and frames per second (FPS) of 85.7 on a high-performance GPU (RTX 2080TI). Furthermore, to achieve the benefits of lower latency, higher throughput, and improved privacy, we optimized our model for implementation on a popular edge-computing device (Jetson Nano GPU) with the TensorRT network optimizer. We have also performed a comparative analysis of the previous weapon detector with our presented model using different CPU and GPU machines that fulfill the purpose of this work, making the selection of model and computing device easier for the users for deployment in a real-time scenario. The analysis shows that our presented models result in improved mAP scores on high-performance GPUs (such as RTX 2080TI), as well as on low-cost edge computing GPUs (such as Jetson Nano) for weapon detection in live CCTV camera surveillance videos.
Place, publisher, year, edition, pages
MDPI , 2022. Vol. 12, no 12, article id 5772
Keywords [en]
weapon detection, object detection, deep learning, optimization, computer vision
National Category
Computer graphics and computer vision
Identifiers
URN: urn:nbn:se:bth-23495DOI: 10.3390/app12125772ISI: 000817404000001OAI: oai:DiVA.org:bth-23495DiVA, id: diva2:1686090
Note
open access
2022-08-082022-08-082025-02-07Bibliographically approved