A lightweight YOLOv5 for real-time dangerous weapons detection

Aicha Khalfaoui, Abdelmajid Badri, Ilham El Mourabit


Deep neural networks are currently employed to detect weapons, and although these techniques provide a high level of accuracy, it still suffers from large weight parameters and a slow inference speed. When it comes to real-world applications, such as weapon detection, these methods are often not suitable for deployment on embedded devices. Because of the huge number of parameters and poor efficiency. The most recent object detection technique, which belongs to the YOLOv5 class, is commonly used for detecting weapons. However, it faces some difficulties such as high computational parameters and an unfavorable detection rate. to solve these shortcomings. an enhanced lightweight Yolov5s approach is suggested. Which consists of a combination of YOLOv5 and GhostNet modules. To evaluate the efficacy of the suggested technique, a set of experiments was performed on the Sohas weapon dataset., which is commonly used as a reference dataset in the field. Compared to the original YOLOv5, the results indicate a slight increase in the proposed model's mean Average Precision (mAP). Furthermore, there has been a reduction of 2.7 in GFLOPs and weights, and the number of model parameters has decreased by 1.42.


Deep learning; Embedded devices; GhostNet; Weapons detection; YOLOv5

Full Text:


DOI: http://doi.org/10.11591/ijai.v13.i2.pp1838-1844


  • There are currently no refbacks.

Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.

IAES International Journal of Artificial Intelligence (IJ-AI)
ISSN/e-ISSN 2089-4872/2252-8938 
This journal is published by the Institute of Advanced Engineering and Science (IAES) in collaboration with Intelektual Pustaka Media Utama (IPMU).

View IJAI Stats