交通标志检测中的数据打包

摘要

不同目标检测模型的标注格式是不一样的。比较评估多种模型时,需要对标注文件进行数据打包,获得对应模型的标注数据格式。本文提出一种简便的数据打包方案,即利用通用的VOC格式作为转换桥梁,只需要少量编程,就可以转换为大部分主流检测模型的标注数据格式。首先创建对应的VOC文件夹结构;再根据图片名称遍历整个原标注文件,同时调整图片类型与尺寸,并根据调整后的图片尺寸比率转换标注框数据;获得对应的VOC格式标注文件。实验将交通标志检测数据集GTSDB的标注文件转换为YOLO格式,结果表明该数据打包方案能够获得准确的标注文件信息,方便对比目标检测模型的精度及速度等指标。