-
公开(公告)号:CN118736519A
公开(公告)日:2024-10-01
申请号:CN202410734978.4
申请日:2024-06-07
Applicant: 燕山大学
IPC: G06V20/56 , G06V10/774 , G06V10/40 , G06V10/766 , G06V10/80
Abstract: 本发明公开了一种自动驾驶图像目标检测方法,属于自动驾驶图像的目标检测技术领域,包括以下步骤:对数据集进行格式转换;将数据集按比例划分为训练集、验证集和测试集;对Backbone进行改进,采用极化自注意力改进骨干网络中的C3结构,分别对网络的底层和顶层特征图进行下采样操作,得到的特征再与普通卷积提取的特征相结合;对Neck进行改进,将颈部结构中的标准卷积改进为轻量型卷积GSConv,将颈部结构中的C3改进为VoV‑GSCSP;设置训练要用的相关参数,进行迭代训练;选择验证集损失函数较小的模型训练权重载入到网络中,获得网络的检测性能。本发明能够提高自动驾驶图像的目标检测精度和实时性。