Pytorch+yolov4做马路边线检测（附代码和数据集）

Pytorch+yolov4做马路边线检测

1.环境版本
Python : 3.7.3
torch : 1.8.0
torchvision : 0.9.0
Cuda : 11.2
2.数据集：
数据集均是我自己采集自己标定的，数量不多但是也够用了。附百度云链接
链接：https://pan.baidu.com/s/1muN1aC3lJxDnRsHyXfnlzw
提取码：e15z
3.项目代码：
项目代码是从github上下载的，数据集是仿照的VOC数据格式，稍微做了一些改动，也是百度云链接吧，毕竟代码是别人的，不好意思再发git。
链接：https://pan.baidu.com/s/1Pwv62b1QK2MNR2tCf9BBMQ
提取码：nhwq
4.运行代码
按照上面的步骤下载yolov4的项目并打开，然后先找到VOC的yolov4-pytorch-LBX\VOCdevkit\VOC2007这个路径，运行里面的voc2yolo4.py。然后回到项目根目录，运行voc_annotation.py，运行完之后，观察是否出现了2007_train.txt文件中是否有如下内容，路径是否是自己存放的路径。

如果没有问题，则可以运行train.py。训练完毕后，运行predict.py查看结果。