交通标志检测在自动驾驶和交通监控中是一个重要的问题。目标检测技术极大地促进了这一过程的自动化。本文实现基于DETR目标检测模型识别交通标志。
Tiny LISA交通标志检测数据集
本文数据集使用Kaggle上提供的Tiny LISA交通标志检测数据集(https://www.kaggle.com/datasets/mmontiel/tiny-lisa-traffic-sign-detection-dataset)。它是LISA交通标志检测数据集的一个子集。
这个数据集中只有900张图像，附带了一个CSV标注文件。由于训练代码库需要XML格式的标注，所以需要进行预处理步骤，将CSV转换成XML。
Tiny LISA数据集包含了9中不同的交通标志类别：
- keepRight（靠右行驶）
- merge（并道）
- pedestrianCrossing（人行横道）
- signalAhead（前方信号）
- speedLimit25（限速25）
- speedLimit35（限速35）
- stop（停车）
- yield（让行）
- yieldAhead（前方让行）
原始数据集结构如下：
db_lisa_tiny/
├── annota
