Object Detection: Harmful Weapons Detection using YOLOv4

Closed-circuit television (CCTV) is essential in the security industry by providing surveillance, monitoring activities, recording incidents, and storing evidence. Research and developments have been performed to ameliorate its application to meet the ever-changing security landscape. This paper presents a revolutionary method to enhance the application of CCTVs in Malaysia. The purpose of this study is to develop an Artificial Intelligence (AI) based weapons detection that helps people in identifying violent crimes that are currently happening. This study focuses on detecting harmful weapons such as handguns and knives using the custom trained object detection model that has been trained using the YOLOv4 Darknet framework. Two sets of training have been done to test the effectiveness of this system. The first training was done on a single class custom object detection model while the second was done on a multiple class custom object detection model. Based on the results obtained, the single class object detection only managed to achieve 66.67% to 77.78% accuracy on average whilst the multiple class object detection managed to achieve up to 100% accuracy on most of its input images.