基于深度学习用于无人驾驶的交通目标检测方法

    公开(公告)号:CN115937636A

    公开(公告)日:2023-04-07

    申请号:CN202211703954.X

    申请日:2022-12-29

    Abstract: 本发明提供一种基于深度学习用于无人驾驶的交通目标检测方法,在现有的成熟模型YOLOv5基础上,通过在SPP模块前加入一种改进的卷积与自注意力机制融合的ACmix模块、增加多尺度目标检测层,接下来下载并处理BDD100K数据集,构建用于模型训练的训练集、验证集和测试集,最后将BDD100K数据集送入构建好的基于YOLOv5改进的交通目标检测模型进行模型的训练、测试、评估;在模型构建阶段,引入的ACmix模块更有利于提取目标特征。在训练阶段,将送入的数据集中不包含交通目标的图像进行删除,以免干扰模型的训练,加速网络收敛。在评估阶段,通过调整模型的宽度和深度,优化模型的精确度和速度,以满足实际应用的需求。

Patent Agency Ranking