交通数据集格式转换脚本

DAIR-V2X

  将 DAIR-V2X 数据集转为COCO、YOLOv5、PP-Vehicle格式

用法

  将image、label、data_info.json放在同级目录下，输入以下命令：

# yolov5.py pp_vehicle.py 同理
python coco.py --data_dir=your_dataset_dir

TT-100K

  将 TT-100K 数据集转为COCO格式

用法

  下载 2021 版 TT-100K 并将 2016 版 annotations.json 放至同级（2021版的标注是2016版的补充，但不能单独使用），输入以下命令：

python coco.py --data_dir=your_dataset_dir