为了训练自己的车道线检测模型,由于tusimple数据集场景较为单一,因此自己标注了一个数据集
模型:lanenet
数据集格式:tusimple
标注工具:labelme
第一步:原始数据集标注
1、使用labelme进行数据标注
使用指令进行安装labelme
pip install labelme
2、在环境下使用指令进行启动labelme
labelme
3、进入界面后选择图片,右键选择Create lineStrip选项进行线段标记
第二步:json数据转换
1、使用指令将json文件转换成标记数据
labelme_json_to_dataset ****.json
转换后会生成如下的文件:
2、批量处理需要编写相应的脚本文件:
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
if __name__ == '__main__':
path = "./images"
for scan_dir in ['night_drive_merchandise']:
json_data = os.path.join