The application of machine learning models in the analysis of helmet-related images has yielded remarkable results in identifying and classifying helmet-wearing behaviours. Previous research has employed several pretrained models to predict proper or improper helmet use, achieving high accuracy on the Helmet Wearing Image Dataset (2024), a newly introduced dataset designed to enhance classification capabilities. This study aims to improve prediction performance on helmet datasets by leveraging state-of-the-art deep learning models and ensemble techniques. Using ResNet-50, MobileNetV2, and EfficientNet-B0 models, the proposed EnsemHelmet Framework uses a soft voting ensemble to optimise the classification results, achieving an outstanding accuracy of 99.24% on the experimental dataset. The results demonstrate the potential of ensemble learning to achieve high performance. This study not only improves the accuracy of the helmet-wearing recognition system but also highlights the effectiveness of ensemble techniques in optimizing performance on real-world datasets.
Tạp chí khoa học Trường Đại học Cần Thơ
Khu II, Đại học Cần Thơ, Đường 3/2, Phường Ninh Kiều, Thành phố Cần Thơ, Việt Nam
Điện thoại: (0292) 3 872 157; Email: tapchidhct@ctu.edu.vn
Chương trình chạy tốt nhất trên trình duyệt IE 9+ & FF 16+, độ phân giải màn hình 1024x768 trở lên