一种基于YOLO v4的路面信息识别方法及系统

    公开(公告)号:CN113255524A

    公开(公告)日:2021-08-13

    申请号:CN202110583167.5

    申请日:2021-05-27

    摘要: 本发明提供了一种基于YOLO v4的路面信息识别方法及系统,使用训练集对基于YOLO v4的路面信息识别模型进行训练,所述基于YOLO v4的路面信息识别模型采用DenseNet网络作为基础网络,所述DenseNet网络中的每个Dense Block连接注意力模型;网络输入滤波器采用5*5的卷积核,对输入的图像进行降维;网络的激活函数为R‑Relu函数;在Dense Block之间嵌入FPN网络;并采用soft‑NMS算法对路面信息设定阈值参数;使用训练好的路面信息识别模型识别出待识别图片或者视频中的路面信息,相比原YOLO v4,识别准确率更高、检测速度更快,也改善了小目标路面信息的检测效果。