《Lane Detection and Classification using Cascaded CNNs》
本文为效果复现过程,详细理论知识请阅读原文。提到的方法主要是两个CNN网络,实现了车道线检测和车道线分类。效果还不错,实时效果还未测。
https://arxiv.org/pdf/1907.01294.pdf
作者提供的源码,请参考https://github.com/fabvio/Cascade-LD
我复现的代码,百度云链接:https://pan.baidu.com/s/1pDleEN0RTfHCNQ9zQGTK5A
提取码:owfd
win10系统、CPU
安装有thorch、numpy、matplotlib等库
复现时间:2020.06.23
我只是将作者的main.ipynb代码提取出来,改成test.py文件。
作者提供的有训练权重,效果复现只需要运行main.ipynb即可。
或者运行test.py也行。
python test.py
效果展示:
Tusimple数据集测试,效果很好。
![](https://i-blog.csdnimg.cn/blog_migrate/f0c10bfad89b5e813f07785f803e2aa2.png)
随意找了一些数据集测试,效果还不错。
![](https://i-blog.csdnimg.cn/blog_migrate/9266e6620522030c721392cc733e74b9.png)
又找了一些,效果稍差些,误将指示箭头检测成车道线。
![](https://i-blog.csdnimg.cn/blog_migrate/f48243bc7ac19862cf4f63d244fce284.png)
![](https://i-blog.csdnimg.cn/blog_migrate/1a0f6997c0047d7336be81669d159e5c.png)