Comparative Evaluation of Yolo-Based Models for Multi-Class Weapon Detection
Abstract
This study explores the effectiveness of various real-time object detectors for the problem of classifying different weapon types. To that end, the performance of YOLOv9, YOLOv10, YOLOv11, YOLOX, and YOLOR models were evaluated on a balanced dataset of 15,387 images across 5 weapon classes. All models are trained with 5-fold stratified cross-validation and YOLOR achieved the best overall performance with an mAP@0.5 of 0.8654 and mAP@0.5:0.95 of 0.8380. In addition, YOLOR obtained a precision of 0.8550, recall of 0.9570, and F1-score of 0.9210. Results show YOLOR’s suitability for high-precision, critical applications, while YOLOX offers advantages for recall-focused, resource-constrained deployments with a recall of 0.9628 despite a lower precision of 0.6902. The experiments highlighted the potential of real-time AI-based object detectors for enhancing security systems.
Keywords
References
- D. G. Lowe, “Object Recognition from Local Scale-Invariant Features,” 1999.
- N. Dalal and B. Triggs, “Histograms of Oriented Gradients for Human Detection.” [Online]. Available: http://lear.inrialpes.fr
- P. Yadav, N. Gupta, and P. K. Sharma, “A comprehensive study towards high-level approaches for weapon detection using classical machine learning and deep learning methods,” Feb. 01, 2023, Elsevier Ltd.
- N. Hnoohom, P. Chotivatunyu, and A. Jitpattanakul, “ACF: An Armed CCTV Footage Dataset for Enhancing Weapon Detection,” Sensors, vol. 22, no. 19, Oct. 2022.
- S. Yellapragada et al., “CCTV-Gun: Benchmarking Handgun Detection in CCTV Images,” Jul. 2023, [Online]. Available: http://arxiv.org/abs/2303.10703
- Y. Gu, X. Liao, and X. Qin, “YouTube-GDD: A challenging gun detection dataset with rich contextual information,” Mar. 2022, [Online]. Available: http://arxiv.org/abs/2203.04129
- D. Berardini, L. Migliorelli, A. Galdelli, E. Frontoni, A. Mancini, and S. Moccia, “A deep-learning framework running on edge devices for handgun and knife detection from indoor video-surveillance cameras,” Multimed. Tools Appl., vol. 83, no. 7, pp. 19109–19127, Feb. 2024.
- J.-S. Lim, M. Astrid, H.-J. Yoon, and S.-I. Lee, “Small Object Detection using Context and Attention,” Dec. 2019, [Online]. Available: http://arxiv.org/abs/1912.06319
Details
Primary Language
English
Subjects
Image Processing
Journal Section
Research Article
Publication Date
June 30, 2026
Submission Date
August 28, 2025
Acceptance Date
April 9, 2026
Published in Issue
Year 2026 Volume: 68 Number: 1
