Skip to content

效果很好的lanenet网络,主干网络基于bisenetv2并对主干网络做了修改,效果远好于bisnetv2

Notifications You must be signed in to change notification settings

mengpengfei/pytorch-lanenet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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、将代码封装,抽取配置,整理代码

About

效果很好的lanenet网络,主干网络基于bisenetv2并对主干网络做了修改,效果远好于bisnetv2

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages