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文件夹下查看测试结果

相关推荐
IT古董4 小时前
【第五章:计算机视觉-计算机视觉在工业制造领域中的应用】1.工业缺陷分割-(1)工业品缺陷风格基础知识:割任务定义、数据集介绍
计算机视觉·3d·自动驾驶
Mr.Winter`16 小时前
深度强化学习 | 基于SAC算法的动态避障(ROS C++仿真)
人工智能·深度学习·神经网络·机器人·自动驾驶·ros·具身智能
酌量17 小时前
路径平滑优化详解(二次规划): 数学建模与目标函数推导
经验分享·笔记·学习·机器人·自动驾驶
青云交17 小时前
Java 大视界 -- Java 大数据在智慧交通自动驾驶仿真与测试数据处理中的应用
java·大数据·自动驾驶·数据存储·算法优化·智慧交通·测试数据处理
箫乾1 天前
第78篇:AI+交通:自动驾驶、智能交通管理与物流优化
人工智能·机器学习·自动驾驶
ARM+FPGA+AI工业主板定制专家1 天前
基于JETSON/RK3588机器人高动态双目视觉系统方案
人工智能·机器学习·fpga开发·机器人·自动驾驶
Ray Song1 天前
自动驾驶传感器数据录制过程中的五大系统性挑战
人工智能·自动驾驶·数据录制
智行众维2 天前
自动驾驶的“虚拟驾校”如何炼成?
人工智能·自动驾驶·汽车·智能驾驶·智能网联汽车·智能驾驶仿真测试·智驾系统
ARM+FPGA+AI工业主板定制专家2 天前
基于Jetson+FPGA+GMSL+AI的自动驾驶数据采集解决方案
人工智能·机器学习·自动驾驶
松果财经3 天前
千亿级赛道,Robobus 赛道中标新加坡自动驾驶巴士项目的“确定性机会”
人工智能·机器学习·自动驾驶