1、数据集简介
随着国家惩治行车不戴头盔违法行为的力度不断加大,双轮车(电动车与摩托车)头盔检测任务也越来越重要。双轮车佩戴头盔检测数据集(two wheeler helmet dataset,TWHD)收集了来自开源数据OSF dataset、bike helmet dataset的数据进行重新标注,并添加了网络爬虫所收集到的图像,按照双轮车与驾乘人员整体、未戴头盔的人头、戴头盔的人头进行定位与分类标注。主要包括:
- 图像与标注数量各5448张
- OSF dataset随机抽取4710张并重新标注
- bike helmet dataset与网络图像738张
2、数据集内容
数据集已经按照VOC格式进行整理,可以直接通过labelImg读取打开查看相关的标注信息。图像分辨率不一,总体在1080P以下,不区分训练集和测试集,总计5448张图像样本与标注xml文件。数据集标注共三个类别:
- two_wheeler,二轮车主体,黄色
- helmet,佩戴头盔的头部,绿色
- without_helmet,未戴头盔的头部,红色