BOSCH Small Traffic Lights Dataset
收藏github2024-04-02 更新2024-05-31 收录
下载链接:
https://github.com/berktepebag/Traffic-light-detection-with-YOLOv3-BOSCH-traffic-light-dataset
下载链接
链接失效反馈官方服务:
资源简介:
BOSCH Small Traffic Lights Dataset是一个用于训练YOLOv3模型以实时检测交通灯的数据集。该数据集包含约6GB的数据,包括5093张图像,分布在7个不同的文件夹中。
The BOSCH Small Traffic Lights Dataset is designed for training the YOLOv3 model to detect traffic lights in real-time. This dataset comprises approximately 6GB of data, including 5093 images distributed across 7 different folders.
创建时间:
2019-10-18
原始信息汇总
数据集概述
数据集名称
- Bosch Small Traffic Lights Dataset
数据集来源
- 下载链接:Bosch Small Traffic Lights Dataset
- 下载要求:需注册后通过邮件接收下载代码。
数据集大小
- 约6GB
数据集结构
- 包含7个不同类别的交通灯图像。
- 共有5093张图像,适合训练交通灯检测模型。
数据集处理
- 使用YOLOv3进行对象检测训练。
- 数据集通过Python脚本转换为Pascal-VOC格式,以便YOLOv3使用。
- 数据集中的图像和标签文件分布在不同的文件夹中,需进行整理以适应YOLOv3的训练要求。
数据集训练
- 使用YOLOv3的预训练模型进行迁移学习。
- 训练过程中需要调整YOLOv3配置文件以适应数据集的类别和大小。
- 训练完成后,模型可用于图像、视频和Carla模拟器的实时交通灯检测。
数据集应用
- 模型训练完成后,可用于NVIDIA Jetson Nano等小型系统的实时交通灯检测。
- 通过ROS(机器人操作系统)集成,模型可在Carla模拟器中进行交互式应用。
数据集准备步骤
-
下载与解压
- 注册并下载数据集。
- 使用7-zip解压数据集。
-
数据集整理
- 创建文件夹结构:
traffic_light_images,traffic_light_xmls,traffic_light_labels。 - 将所有图像移动到
traffic_light_images文件夹。
- 创建文件夹结构:
-
标签转换
- 使用
bosch_to_pascal.py脚本生成Pascal-VOC格式的XML标签文件。 - 使用
bosch_voc_to_yolo_converter.py脚本将Pascal-VOC标签转换为YOLO格式。
- 使用
-
训练准备
- 配置YOLOv3的
yolov3-tiny-bosch.cfg文件,调整类别数和过滤器大小。 - 使用预训练的
darknet53.conv.74权重开始训练。
- 配置YOLOv3的
-
模型测试
- 训练完成后,使用
./darknet detector demo命令测试模型在视频上的表现。
- 训练完成后,使用
数据集使用环境
- 支持CUDA的NVIDIA GPU。
- 操作系统:Ubuntu。
- 依赖软件:ROS, Carla模拟器。
数据集应用场景
- 实时交通灯检测系统。
- 自动驾驶车辆中的交通灯识别。
- 交通监控系统中的交通灯状态检测。
搜集汇总
数据集介绍

构建方式
BOSCH Small Traffic Lights Dataset的构建过程基于真实交通场景的采集与标注。数据集包含了5093张图像,涵盖了多种交通灯状态,如红灯、绿灯、黄灯及其不同方向的变化。每张图像均通过精确的边界框标注,确保交通灯的位置和状态信息准确无误。此外,数据集还提供了Python脚本,用于将标注数据转换为Pascal-VOC格式,便于后续的深度学习模型训练。
特点
该数据集的特点在于其多样性和高质量标注。数据集不仅包含了丰富的交通灯状态,还涵盖了不同光照条件、天气状况和交通场景,确保了模型的泛化能力。每张图像的标注信息详细,包括交通灯的类型、位置和状态,为模型训练提供了坚实的基础。此外,数据集的规模适中,既保证了训练的充分性,又避免了过大的计算负担。
使用方法
使用BOSCH Small Traffic Lights Dataset进行模型训练时,首先需要将数据集转换为YOLOv3所需的格式。通过提供的Python脚本,将Pascal-VOC格式的标注转换为YOLO格式。随后,利用YOLOv3框架进行模型训练,通过调整配置文件和数据路径,确保模型能够正确读取和处理数据。训练完成后,模型可以用于实时交通灯检测,支持图像、视频以及Carla模拟器中的测试。
背景与挑战
背景概述
BOSCH Small Traffic Lights Dataset是由博世公司(Bosch)与海德堡大学(University of Heidelberg)合作创建的一个专注于交通灯检测的数据集。该数据集于2017年发布,旨在为自动驾驶和智能交通系统提供高质量的交通灯检测数据。数据集包含5093张图像,涵盖了多种交通灯状态和类型,如红灯、绿灯、黄灯及其不同方向的变化。该数据集的发布为交通灯检测算法的研究提供了重要的基准,推动了实时目标检测技术的发展,尤其是在自动驾驶领域。通过使用该数据集,研究人员能够训练和验证基于深度学习的交通灯检测模型,如YOLOv3,从而提升自动驾驶系统在复杂交通环境中的感知能力。
当前挑战
BOSCH Small Traffic Lights Dataset在应用过程中面临多重挑战。首先,交通灯检测本身具有较高的复杂性,尤其是在不同光照条件、天气变化和遮挡情况下,模型需要具备较强的鲁棒性。其次,数据集的构建过程中,交通灯的多样性和标注的准确性是关键问题,确保每一类交通灯状态和方向都能被准确标注,需要大量的人工干预和验证。此外,数据集的预处理和转换过程也较为复杂,例如将Pascal VOC格式的标注转换为YOLO格式,涉及到大量的脚本编写和调试工作。最后,尽管数据集提供了丰富的图像,但在实际应用中,模型的实时性和计算效率仍然是需要解决的难题,尤其是在资源受限的嵌入式设备上运行检测模型时,如何在保证精度的同时提升速度,是研究人员需要持续探索的方向。
常用场景
经典使用场景
BOSCH Small Traffic Lights Dataset 在交通信号灯检测领域具有广泛的应用,尤其是在自动驾驶和智能交通系统中。该数据集通过提供大量标注的交通信号灯图像,为研究人员和开发者提供了一个可靠的基准,用于训练和评估实时目标检测模型。经典的使用场景包括在复杂的城市环境中,利用YOLOv3等深度学习模型进行交通信号灯的实时检测与分类,确保自动驾驶车辆能够准确识别并响应交通信号。
实际应用
在实际应用中,BOSCH Small Traffic Lights Dataset 被广泛用于自动驾驶汽车的感知系统中。通过结合YOLOv3等实时目标检测算法,该数据集能够帮助车辆在复杂的城市道路环境中准确识别交通信号灯,确保行车安全。此外,该数据集还被用于智能交通系统的开发,通过实时监控交通信号灯的状态,优化交通流量管理,减少拥堵和事故的发生。
衍生相关工作
基于BOSCH Small Traffic Lights Dataset,许多经典的研究工作得以展开。例如,研究人员利用该数据集开发了基于YOLOv3的交通信号灯检测模型,并在Carla仿真平台上进行了验证。此外,该数据集还催生了多个开源项目,如结合ROS和YOLO的实时检测系统,进一步推动了自动驾驶技术的发展。这些工作不仅提升了交通信号灯检测的精度和效率,还为相关领域的研究提供了宝贵的参考和工具。
以上内容由遇见数据集搜集并总结生成



