加油
背景
- 由于项目需求，数据集需要扩展，包括车辆行人等类别，于是找到KITTI数据集，现需将其转换为voc格式
- KITTI数据集下载：Object Detection Evaluation 2012
- 参考博客（所有内容均来自这个博客，只是针对yolo需要的格式进行截取及小改，感谢博主！）:SSD: Single Shot MultiBox Detector 训练KITTI数据集（1）
转换步骤
参考博客中下载好数据集，并建立好文件夹后：
1. 转换KITTI类别
- 我需要的类别是车辆和行人，所以跟博客中一样，未修改。需要注意的是博客中的脚本是直接在原始文件上修改并覆盖的，所以这边我加了几行代码备份了一下原始文件。脚本：modify_annotations_txt.py
2. 转换txt标注信息为xml格式
3. 生成训练验证集和测试集列表
- 博客中生成的txt中只有图片名称，我这边由于yolo格式需要加入了图片绝对路径
- 脚本：create_train_test_txt.py