一:环境配置
1.python3.8, cuda12.1 pytorch2.2.2
2.代码下载:链接
二:训练与测试
1.数据集下载:
CULane:https://xingangpan.github.io/projects/CULane.html(下载完放在CULane目录)
tusimple:https://github.com/TuSimple/tusimple-benchmark/issues/3
2.config,log配置
打开config-->culane.py中dataset='CULane' log_path = './log',配置数据集和日志
3.开始训练
python train.py --config ./configs/culane.py --test_model culane_18.pth
4.测试
1.准备测试数据放在autodl-tmp/Lane-Detection/CULane/list/test_split/my_test.txt中,选好模型开始测试python demo.py --config configs/culane.py --test_model culane_18.pth