github源码:github
源码链接,已上传百度云链接:https://pan.baidu.com/s/10jv__0CbKdd-kUKpBP6JiQ
提取码:hroa
windows,GPU,CUDA(推荐10.0,cudnn百度查询相关匹配版本及配置教程,
pytorch1.0.1(pip无法下载 百度云链接链接:https://pan.baidu.com/s/1zlOpwnrcKhFUtD2AzWVMTA
提取码:o5a5)
- 下载源码,pycharm打开如图
- 内部已经有原作者预训练的模型,自己选择添加测试数据
- 几处路径更改
parameters.py
line22 mode=? 0测试文件夹内全部图片,1测试视频,2测试单张图片
line12 save_path 结果保存路径
line13 model_path 调用的模型路径
line53 train_root_url 训练集路径
line54 test_root_url 测试集路径
上传的源码内已更改
data_loader.py
line50… 调用数据集相关json文件路径,上传的源码内已经更改
test.py
line 44 选择预训练的模型
line 66 测试视频文件路径
line 87 测试图片文件路径(图片尺寸要求1270*720)
效果截图
想要自己训练去下载TuSimple数据集TuSimple
根据自己显卡配置调整batch_size大小,显存多就往大了调
其他相关训练的模型保存路径配置原作者那有说(顶部github链接)