tusimple数据集处理

一、下载数据集

1、官方下载地址 https://github.com/TuSimple/tusimple-benchmark/issues/3，数据集很大，下载速度超级慢。

2、参考https://blog.csdn.net/flana/article/details/105098470

百度网盘下载地址链接: https://pan.baidu.com/s/1iyEMu0tcKaVX8nv0zBnSKw 提取码: fccj，网盘中只有zip压缩包，json文件到官网复制粘贴即可。

下载完后文件目录结构如下：

目前只用了train_set.zip、test_set.zip和几个json文件，其它几个压缩包没用到。

二、处理数据

参考：https://blog.csdn.net/qq_40900196/article/details/102840850

1、解压train_set.zip, 解压后如下：

2、根据json文件转换训练集，生成图片文件夹gt_image、gt_binary_image、gt_instance_image 以及文本文件 train.txt

python tools/generate_tusimple_dataset.py --src_dir f:/Data/tusimple/train_set

处理过程如下图所示：

注意：处理过程时间较长，且有一段时间没有打印信息，个人猜测在写txt文件

处理完成后文件目录结构如下图所示：

3. 生成test.txt 以及val.txt, 将标注格式转换成TFRecord

注意：网上很多博客是针对旧版本的代码写的，如果还按照旧版本的去操作，会报错。报的错例如：

最新版代码对应的操作：

1) 修改 config文件夹下tusimple_lanenet.yaml文件

修改内容如下图所示，绿色的为原作者的数据路径，将数据集路径修改为自己的即可，具体路径参考上一步。

2）执行 python tools/make_tusimple_tfrecords.py

执行到下图状态时要等待很久。

执行完成后如下图所示，提示Generating testing example tfrecords complete

执行完成后文件结构如下图所示：

三、训练

训练相关的参数比如batch_size等，参考tusimple_lanenet.yaml文件。

执行训练脚本：python tools/train_lanenet_tusimple.py

这里遇到了错误，提示No module named 'trainner

解决方法在train_lanenet_tusimple.py文件最开始的地方添加下面两行：

import syssys.path.append(r'D:\Projects\LaneDetect\lanenet-lane-detection')

上图中的路径根据自己的实际情况进行修改。

再次运行python tools/train_lanenet_tusimple.py，可以跑起来了，我这台电脑没有GPU，我把batch_size调到了1 , 才跑起来。

这里只是把脚本执行起来，真正的训练应该是要GPU的，否则的话速度慢不说，而且训练出的结果肯定也不理想。

LaneNet车道线检测tusimple数据集处理相关推荐

车道线识别 tusimple 数据集介绍
1.tusimple 数据集介绍标注json 文件中每一行包括三个字段 : raw_file : 每一个数据段的第20帧图像的的 path 路径 lanes 和 h_samples 是数据具体的标注 ...
Lanenet 车道线检测网络模型学习（论文解读+官方模型）
本文讲解的是用于车道线检测的一个网络结构叫lanenet, 转载请备注,多谢哈|! 2018.2发表出来的,文章下载地址:https://arxiv.org/abs/1802.05591 github ...
自动驾驶——车道线检测相关数据集整理
自动驾驶视觉方向的分支车道线检测需要的相关数据集暂时没有好好整理,先作个记录: 1.KITTI KITTI数据集由德国卡尔斯鲁厄理工学院和丰田美国技术研究院联合创办,主要于算法评测. KITTI数据 ...
Lanenet车道线检测 --运行别人代码（1）
2018.2发表出来的,文章下载地址:https://arxiv.org/abs/1802.05591 github上代码:https://github.com/MaybeShewill-CV/lan ...
车道线识别--TuSimple数据集
TuSimple数据集:Github网址网盘链接提取码:gquc
3.1.1 车道线检测
更多内容,请关注: github:Autopilot-Updating-Notes gitee: Autopilot-Updating-Notes 车道线检测是一种环境感知应用,其目的是通过车载相机或 ...
车道线检测参考学习资料
一.GitHub: https://github.com/amusi/awesome-lane-detection(全) https://github.com/ChengZhongShen/Advan ...
CVPR2022车道线检测Efficient Lane Detection via Curve Modeling
分享前段时间看的一篇车道线检测方向的新工作,也是中了最近公开结果的2022CVPR,是上海交大.华东师大.香港城市大学和商汤科技合作完成的,代码已经开源.关于车道线检测任务,我之前也分享过几篇文章: ...
车道线检测CLRNet算法复现在Tusimple数据集测试demo
0 写在前面分享最近在车道线检测方面的工作,详细跟大家说下我的配置环境和过程,欢迎自动驾驶的朋友来交流学习,助力自动驾驶早日落地. 分享一篇今年CVPR车道线检测方向的新工作--CLRNet: Cr ...

LaneNet车道线检测tusimple数据集处理

tusimple数据集处理

一、下载数据集

二、处理数据

三、训练

LaneNet车道线检测tusimple数据集处理相关推荐

最新文章

热门文章