基于YOLOv4的交通标志和道路病害识别

博主使用YOLOv4进行了交通标志和井盖的识别任务,训练数据集为tt100k。训练中遇到断电导致最高mAP为74%,测试结果显示部分图标识别错误。同时,通过迁移学习识别井盖,并指出识别地面裂缝可能需要大量数据集和网络结构调整。训练环境为Win10,显卡为Nvidia GeForce 2080 Super,内存限制导致分辨率降低。
摘要由CSDN通过智能技术生成

这周用yolov4干了两件事情,一个是训练tt100k交通标志数据集并测试,一个是用tt100k的图片标注出井盖,然后训练测试。

 

1. 训练tt100k交通标志数据集并测试

所用的机器:64G RAM; i7-10代;Nvidia GeForce 2080 super 8G

环境:CUDA 10.2 ; cudnn-8.0.2 ; python 3.7; openCV 3.4 ; yolov4

 

训练了两天左右,maxbatch设置为90000,训练到45000左右的时候意外断电,最高map为74%。

下图为博主用Nvidia Titan 11G 训练完成之后的结果

下面图片是我用训练完(一半)的模型测试的结果

评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值