无人驾驶目标检测方法、装置、设备及介质与流程

技术特征：
1.一种无人驾驶目标检测方法，其特征在于，包括以下步骤：获取路面图像的样本集，对所述样本集进行预处理，并将经过预处理后的样本集分为训练集、验证集和测试集；其中，所述路面图像是由无人驾驶汽车上的图像采集设备采集到的；构建改进yolov5模型；其中，改进主要包括：对bottleneckcsp的模块数量进行修改；在spp结构后引入协同注意力机制；在panet中采用双线性插值法进行上采样；在原有网络结构三层检测层的基础上，增加小目标检测层，以对小目标物体进行检测；将panet中的低层特征与高层特征进行融合；在panet中使用基于分离的卷积运算；采用ciouloss作为目标检测任务的回归损失函数；在预测阶段利用diou作为nms的评判标准；将所述训练集和所述验证集输入到所述改进yolov5模型中进行训练，得到训练好的改进yolov5模型；将所述测试集输入到所述训练好的改进yolov5模型中，对所述测试集中的每一图像进行检测，得到目标检测结果。2.根据权利要求1所述的无人驾驶目标检测方法，其特征在于，所述对bottleneckcsp的模块数量进行修改，包括：将yolov5模型的主干网络中bottleneckcsp的模块数量由(
×
3，
×
9，
×
9，
×
3)修改为(
×
2，
×
6，
×
6，
×
2)。3.根据权利要求1所述的无人驾驶目标检测方法，其特征在于，所述在spp结构后引入协同注意力机制，包括：将位置信息嵌入到通道注意力中，以捕获跨通道的信息、方向感知和位置感知信息，使模型更加精准地定位和识别感兴趣的目标；其中，所述协同注意力机制将通道注意力分解为两个一维特征进行编码，沿着不同的空间方向，分别捕获长程依赖与精确的位置信息，编码形成方向感知和位置敏感的注意力特征图来互补地应用到输入特征图，以增强感兴趣目标的表示。4.根据权利要求1所述的无人驾驶目标检测方法，其特征在于，所述将panet中的低层特征与高层特征进行融合，包括：将原始yolov5模型中14与18层的特征融合改为12与26层，并配合增加检测层之后16与23层的特征融合操作；其中，特征融合操作使用concat方法。5.根据权利要求1所述的无人驾驶目标检测方法，其特征在于，所述采用ciouloss作为目标检测任务的回归损失函数具体为：目标检测任务的回归损失函数具体为：其中：
式中，为giou损失，iou表示pb和gt的交并比，pb中心点用b表示，gt中心点用b
gt
表示，ρ2(
·
)表示欧氏距离，c代表pb与gt的最小包围框的最短对角线长度，α是一个正平衡参数，υ代表长宽比一致性的参数，和ω、h分别表示gt和pb的宽度和高度，预测边界框为(predicted box，pb)，真实边界框为(ground truth，gt)。6.根据权利要求5所述的无人驾驶目标检测方法，其特征在于，所述在预测阶段利用diou作为nms的评判标准，包括：diou在iou的基础上考虑了两个边界框中心点的距离，如公式：其中，diou-nms的定义为：nms的定义为：表示预测分数最高的一个预测框，b
i
表示判断是否需要被移除的预测框，s
i
表示分类分数，ε表示nms的阈值。7.一种无人驾驶目标检测装置，其特征在于，包括：数据获取模块，用于获取路面图像的样本集，对所述样本集进行预处理，并将经过预处理后的样本集分为训练集、验证集和测试集；其中，所述路面图像是由无人驾驶汽车上的图像采集设备采集到的；模型构建模块，用于构建改进yolov5模型；其中，改进主要包括：对bottleneckcsp的模块数量进行修改；在spp结构后引入协同注意力机制；在panet中采用双线性插值法进行上采样；在原有网络结构三层检测层的基础上，增加小目标检测层，以对小目标物体进行检测；将panet中的低层特征与高层特征进行融合；在panet中使用基于分离的卷积运算；采用ciouloss作为目标检测任务的回归损失函数；在预测阶段利用diou作为nms的评判标准；模型训练模块，用于将所述训练集和所述验证集输入到所述改进yolov5模型中进行训练，得到训练好的改进yolov5模型；目标检测模块，用于将所述测试集输入到所述训练好的改进yolov5模型中，对所述测试集中的每一图像进行检测，得到目标检测结果。8.一种终端设备，其特征在于，包括处理器、存储器以及存储在所述存储器中且被配置为由所述处理器执行的计算机程序，所述处理器执行所述计算机程序时实现如权利要求1至6中任意一项所述的无人驾驶目标检测方法。9.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质包括存储的计算机程序，其中，在所述计算机程序运行时控制所述计算机可读存储介质所在设备执行如权利要求1至6中任意一项所述的无人驾驶目标检测方法。

技术总结
本发明提供了一种无人驾驶目标检测方法、装置、设备及介质，所述方法包括获取路面图像的样本集，对所述样本集进行预处理，并将经过预处理后的样本集分为训练集、验证集和测试集；构建改进YOLOv5模型；将所述训练集和所述验证集输入到所述改进YOLOv5模型中进行训练，得到训练好的改进YOLOv5模型；将所述测试集输入到所述训练好的改进YOLOv5模型中，对所述测试集中的每一图像进行检测，得到目标检测结果。采用本发明实施例，能够通过在主干网络中引入协同注意力机制，添加小目标检测层，并使用CIoU作为边界框回归的损失函数，能够在原有YOLOv5算法的基础上，提高了检测精度与检测速率，优化了YOLOv5对小目标检测。优化了YOLOv5对小目标检测。优化了YOLOv5对小目标检测。


技术研发人员：林凡 李沐 黄俊贤
受保护的技术使用者：广州杰赛科技股份有限公司
技术研发日：2021.12.02
技术公布日：2022/4/15