Deep Learning-based Weapon Detection using Yolov8

Abstract

Deep learning (DL), a subset of machine learning (ML), has demonstrated remarkable success in image recognition and object detection tasks. This study presents a deep learning-based approach for offline weapon detection using the YOLOv8m architecture. A custom YOLO-formatted dataset was developed, comprising over 10,000 annotated images spanning two weapon categories: guns (all types of firearms) and knives (all types). The model achieved a Mean Average Precision ([email protected]) of 0.852. and [email protected]:0.95 of 0.622, with precision and recall scores of 0.89 and 0.80, respectively. The class-wise evaluation revealed strong detection across both weapons, with [email protected] of 0.871 for knives and 0.831 for guns. Despite occasional false positives and class confusion, the system shows promise for offline weapon detection tasks