Skip to content

Latest commit

 

History

History
37 lines (19 loc) · 1.09 KB

README.md

File metadata and controls

37 lines (19 loc) · 1.09 KB

pytorch-lanenet

效果很好的lanenet网络,主干网络基于bisenetv2并对主干网络做了修改,效果远好于bisnetv2 可直接训练自己的数据应用于生产

inspired by https://github.com/MaybeShewill-CV/lanenet-lane-detection

Using Bisenetv2 as Encoder.

使用步骤:

1、安装pytorch环境,pytorch官网有说明,推荐使用docker

2、生成样本的train.txt和val.txt文件

文件内容:

原始图片 语义分割图 实例分割图

3、修改 script目录下的Convert2LMDB.py main函数中的txt文件路径和生成的lmdb文件名

4、修改lanenet/train.py中的train_dataset_file和val_dataset_file为自己生成的lmdb文件路径

按照自己的需要lanenet/config.py文件中的配置参数

5、执行 python setup.py install

6、执行python lanenet/train.py --lr 0.001 --val True --bs 16 --save ./checkpoints --w1 0.25 --w2 0.25 --w3 0.25 --w4 0.25 --epochs 200

开始训练。

7、训练完成后 修改 lanenet/online_test_video.py 中的模型路径和视频路径,测试视频‘

TODO: 1、将代码封装,抽取配置,整理代码