Automated Detection of Driver and Passenger Without Seat Belt using YOLOv8

Abstract

The issue of traffic accident fatalities is a serious concern on a global scale, and one of the contributing factors is the failure of drivers to adhere to seat belt usage. A notable challenge arises from the limited availability of law enforcement personnel monitoring this particular issue. In this context, there is a compelling need to implement an automated detection system. The development of this system using YOLOv5 has been done. However, there are weaknesses related to the length of training and detection time. Therefore, this paper proposed a new system using the YOLOv8 method to detect drivers and passengers who violate seat belt regulations. The proposed system is divided into three subsystems: windshield detection, passenger classification, and seat belt classification. YOLOv8 is the latest version of the YOLO (You Only Look Once) method and has been proven to provide better performance than previous versions. Furthermore, this paper also compared five YOLOv8 models, namely YOLOv8n, YOLOv8s, YOLOv8m, YOLOv8l, and YOLOv8x. The proposed model is trained and tested using image data collected from several roads in Indonesia. The experiment results show that the YOLOv8s model produced the best mean Average Precision (mAP) of 0.960 for windshield detection. YOLOv8s-cls and YOLOv8l-cls models achieved the same accuracy of 0.8923 for passenger classification. The YOLOv8l-cls model produced the best accuracy of 0.8846 for seat belt classification. In addition, the proposed method can increase mAP and training time for windshield detection compared to YOLOv5.