暂无分类
暂无标签
发布于2020-06-24 12:16 阅读(413) 评论(0) 点赞(5) 收藏(4)
0
1
2
3
4
5
PINet车道线检测是什么,就不多介绍了,可以看看原文,重点讲解一下效果复现过程。
https://arxiv.org/pdf/2002.06604.pdf
可以下载Github上的源码,也可以下载我上传百度云的源码。链接:https://pan.baidu.com/s/1T82o6BSHhrPw759I1vY6mQ
提取码:37ge
注意:Github上的有部分小问题,比如代码缩进等等(可能是作者上传的时候没有注意吧),需要自己修改一下。上传的是我自己测试过的,在服务器上跑的,能出结果。(之前出现过问题,numpy出问题了,一直报错)
Ubuntu18.04、GPU、CUDA(必须有CUDA,没有的话需要改代码)
python3.6.9
基本库都装好了,比如pytorch、numpy等等。
下载源码。下载下来的可能就是如下图这个样子。想要复现效果,还需要自己添加修改些东西。
源码下载下来之后,然后就是下载数据集Tusimple。我已上传百度云,可自行下载。https://pan.baidu.com/s/1htTlI4T194hPPs38k4wztA
提取码:qn0e
我测试的时候,就用了20张图片,所以没必要将全部下载下来。
下载完成之后,在根目录下新建一个文件夹,Tusimple
,里面放的就是测试的图片。Tusimple下还有test_set、clips、0530、1492626047222176976_0这些文件夹,20张图片就放在1492626047222176976_0这个文件夹里面。(这样做是为了顺应作者的源码,当然也可以自己新建图片存放文件夹,记住路径就行了)
测试数据集准备完了之后,就需要修改一些文件路径了。这里面需要修改的有test.py、agent.py、parameters.py
三个文件。
test.py:
63行——改成自己的测试视频路径(不测试视频,就不用管)
82行——改成自己的测试图片路径
192行——两个p.grid_y加上int,强制转换,int(p.grid_y)。(我运行的时候报错了,所以我改了)
agent.py:
245、246、247行:缩进有问题,调整跟244行一样就行了。
parameters.py:
22行:这个是调整测试模式的,0代表测试数据集所有图片;1代表测试视频;2代表测试某一张图片;默认是3,忘了是啥
51、52行:改成自己数据集的路径。
58、59、60行:grid_y、grid_x加int。(我这里报错了,所以我加了)
以上是需要修改的或者需要注意的地方。行数和我写的可能会有出入,看清楚再改。
python test.py
就可以测试效果了。【一定要注意,选好模式,0、1、2、3】检测效果还不错:下面是一张Tusimple数据集的检测图片。
视频没有保存下来,放一张视频图片。
0
1
2
3
4
5
作者:你太美丽
链接: https://www.pythonheidong.com/blog/article/427835/feeb63bddea6dc1bec06/
来源: python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容:(最多支持255个字符)
Copyright © 2018-2019 python黑洞网 All Rights Reserved 版权所有,并保留所有权利。 京ICP备18063182号-1
投诉与举报,广告合作请联系z452as@163.com或QQ3083709327
免责声明:网站文章均由用户上传,仅供读者学习交流使用,禁止用做商业用途。若文章涉及色情,反动,侵权等违法信息,请向我们举报,一经核实我们会立即删除!