LaneNet(2):工程代码复现(pytorch版本)(15分钟上手)

一、下载代码

在github上开源的LaneNet项目数目较少,其中只有基于tensorflow 1.x的项目,这个是基于pytorch版本复现的。

github仓库:https://github.com/IrohXu/lanenet-lane-detection-pytorch

二、数据集准备

上面的代码工程中已经准备好了标注好的数据(只有6张),这里不需要额外准备。

如果想自己标注数据集,可参考我的另一篇文章:tusimple车道线检测 标注自己的数据集

如果想要训练大规模数据,可以前往https://github.com/TuSimple/tusimple-benchmark/issues/3下载公开数据集。

三、训练

3.1 修改训练参数

数据集路径

model/utils/cli_helper.py路径下,修改

训练轮数

model/utils/cli_helper.py路径下,修改

3.2 训练

运行train.py文件

3.3 deeplabv3+训练

默认使用的是ENet作为backbone,如果要使用deeplabv3+结构训练

可以在model/utils/cli_helper.py路径下修改Model_type为DeepLabV3+,重新训练即可。

四、测试

4.1 修改测试参数

修改测试图片路径

model/utils/cli_helper_test.py路径下,修改

4.2 测试

运行test.py文件,在./test_output文件夹下查看测试结果

相关推荐
Uzuki4 小时前
自动驾驶 | 为CarLA添加一辆小米SU7 Part I
自动驾驶·游戏开发·图形学
gaoshengdainzi1 天前
镜片防雾性能测试仪在自动驾驶与无人机领域的创新应用
人工智能·自动驾驶·无人机·镜片防雾性能测试仪
向上的车轮2 天前
NOA是什么?国内自动驾驶技术的现状是怎么样的?
人工智能·机器学习·自动驾驶
深蓝学院2 天前
闭环SOTA!北航DiffAD:基于扩散模型实现端到端自动驾驶「多任务闭环统一」
人工智能·机器学习·自动驾驶
智能汽车人3 天前
自动驾驶---学术论文的常客:nuScenes数据集的使用
人工智能·机器学习·自动驾驶
达柳斯·绍达华·宁4 天前
自动驾驶04:点云预处理03
人工智能·机器学习·自动驾驶
地平线开发者4 天前
【征程 6】工具链 VP 示例为什么能运行
算法·自动驾驶
寻丶幽风5 天前
论文阅读笔记——ReconDreamer
论文阅读·笔记·自动驾驶·3dgs·世界模型·闭环仿真
GIS数据转换器5 天前
在机器人和无人机时代,测绘人的出路在哪里?
大数据·人工智能·信息可视化·机器人·自动驾驶·汽车·无人机
地平线开发者7 天前
精度调优|conv+depth2space 替换 resize 指导
算法·自动驾驶