代码链接：https://github.com/lucastabelini/LaneATT

1. 环境配置

#代码下载
git clone https://github.com/lucastabelini/LaneATT
# 新建虚拟环境
conda create -n laneatt python=3.8 -y
#激活虚拟环境
conda activate laneatt
# 安装torch和torchvision
conda install pytorch==1.6 torchvision -c pytorch
# 再安装必要的python包
pip install -r requirements.txt
# 安装nms
cd lib/nms; python setup.py install; cd -

2.模型训练

python main.py train --exp_name my_r18_culane --cfg cfgs/laneatt_culane_resnet18.yml

3.模型测试及其可视化

python main.py test --exp_name my_r18_culane  --view all