Automated detection of bicycle helmets using deep learning

Abstract

Bicycle helmets are a main measure for injury prevention in case of a crash and are a central variable in transport safety studies. Despite this, helmet use data is only collected sporadically, as the observation of helmet use in traffic by human observers is costly and time-consuming. An automated method for the accurate registration of bicycle helmet use would enable the broad and precise registration of cyclists’ helmet use. In this paper, we develop and test a computer vision-based detection method that can be applied to traffic video data. We record bicycle traffic at two observation sites in Copenhagen, Denmark, and annotate a dataset of 4000 cyclists, registering their helmet use. We then train a state-of-the-art object detection algorithm on the detection of cyclists and helmet use. The developed model has good accuracy in registering active cyclists. For helmet use registration on the test data set, there was an underestimation of 0.52% (algorithm registered helmet use: 50.23%; actual helmet use: 50.75%). Cross-testing the algorithm, i.e., training on one observation site and applying it to another, results in a larger underestimation of bicycle helmet use between 5.28% and 6.31%. Finally, we apply the algorithm to a week of video data from two Copenhagen sites, identifying commuting-related peaks of cyclists and registering helmet use differences between the observation sites. This study shows that computer vision algorithms are a feasible method for the automated detection of bicycle helmet use. Further research needs to be conducted to make the site transfer more robust and to increase accuracy levels.