基于改进YOLOX的实时路况目标检测方法及装置与流程

基于改进yolox的实时路况目标检测方法及装置
技术领域
1.本公开属于人工智能技术领域，具体涉及一种基于改进yolox的实时路况目标检测方法及装置。


背景技术：

2.随着深度学习和神经网络的发展，计算机视觉得以飞速发展。在此领域，目标检测与识别技术被广泛研究并应用于实践，给人们生活带来极大便利。例如应用于自动驾驶上，可以自动地识别行车路上的行人、障碍物和交通信号，可以有效为驾驶者提供安全预警。然而在许多目标检测工作中存在以下问题：一、目标远近不同，尺寸各异，不能有效完全检测和识别；二、背景复杂，干扰因素多，如拍摄角度，光照变化、相似目标、物体遮挡等问题，容易导致误判，不利于检测。


技术实现要素：

3.本公开旨在至少解决现有技术中存在的技术问题之一，提供一种基于改进yolox的实时路况目标检测方法及装置。
4.本公开的一方面，提供一种基于改进yolox的实时路况目标检测方法，所述方法包括：
5.采集实时路况目标的数据集，并对数据集进行预处理；
6.按照预设的比例将预处理后的数据集划分为训练集和测试集；
7.建立改进yolox网络结构：在残差分支中额外引出特征，经过eca通道注意力后与另一分支的特征叠加，每个提取到的特征图都会有多个通道，各个通道上的特征通过叠加就可以得到完整的物体特征；
8.利用所述训练集对所述改进yolox网络结构进行训练，得到实时路况目标检测模型；
9.使用所述实时路况目标检测模型对实时路况目标进行检测。
10.在一些实施方式中，所述eca通道注意力机制如下：
11.对重要的特征分量所在通道赋予加大权重值，对非重要的特征分量所在通道赋予较小权重，然后求加权和。
12.在一些实施方式中，所述重要的特征分量的权重值范围为0.8～0.95，所述非重要的特征分量的权重值范围为0.1～0.25。
13.在一些实施方式中，所述建立改进yolox网络结构，还包括：
14.yolox网络结构的各模块之间采用交叉融合策略，以使得浅层特征中物体特征信息和深层特征中物体特征信息互补。
15.本公开的另一方面，提供一种基于改进yolox的实时路况目标检测装置，所述装置包括：
16.采集模块，用于采集实时路况目标的数据集，并对数据集进行预处理；
17.划分模块，用于按照预设的比例将预处理后的数据集划分为训练集和测试集；
18.建立模块，用于建立改进yolox网络结构：在残差分支中额外引出特征，经过eca通道注意力后与另一分支的特征叠加，每个提取到的特征图都会有多个通道，各个通道上的特征通过叠加就可以得到完整的物体特征；
19.训练模块，用于利用所述训练集对所述改进yolox网络结构进行训练，得到实时路况目标检测模型；
20.检测模块，用于使用所述实时路况目标检测模型对实时路况目标进行检测。
21.在一些实施方式中，所述eca通道注意力机制如下：
22.对重要的特征分量所在通道赋予加大权重值，对非重要的特征分量所在通道赋予较小权重，然后求加权和。
23.在一些实施方式中，所述重要的特征分量的权重值范围为0.8～0.95，所述非重要的特征分量的权重值范围为0.1～0.25。
24.在一些实施方式中，所述建立模块，具体还用于：
25.yolox网络结构的各模块之间采用交叉融合策略，以使得浅层特征中物体特征信息和深层特征中物体特征信息互补。
26.本公开的另一方面，提供一种电子设备，包括：
27.一个或多个处理器；
28.存储单元，用于存储一个或多个程序，当所述一个或多个程序被所述一个或多个处理器执行时，能使得所述一个或多个处理器实现根据前文记载的所述的方法。
29.本公开的另一方面，提供一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时能实现根据前文记载的所述的方法。
30.本公开的基于改进yolox的实时路况目标检测方法及装置，在残差分支中额外引出特征，经过eca通道注意力后与另一分支的特征叠加，每个提取到的特征图都会有多个通道，各个通道上的特征通过叠加就可以得到完整的物体特征，可以有效提高实时路况目标检测精度和检测效率。在自制交通标志数据集上市进行测试，改进后的yolox算法在测试集上的检测精度和召回率都有明显提升。
附图说明
31.图1为本公开一实施例的基于改进yolox的实时路况目标检测方法的流程图；
32.图2为本公开另一实施例的改进yolox网络模型的整体图；
33.图3为本公开另一实施例的eca模块的结构图；
34.图4为本公开另一实施例的改进yolox网络模型采用交叉融合策略的示意图；
35.图5为本公开另一实施例的基于改进yolox的实时路况目标检测装置的结构示意图。
具体实施方式
36.为使本领域技术人员更好地理解本公开的技术方案，下面结合附图和具体实施方式对本公开作进一步详细描述。
37.本实施例的一方面，如图1所示，涉及一种基于改进yolox的实时路况目标检测方
法s100，所述方法s100包括：
38.s110、采集实时路况目标的数据集，并对数据集进行预处理。
39.s120、按照预设的比例将预处理后的数据集划分为训练集和测试集。
40.具体地，在上述步骤s110中，可以采集10000张实时路况目标的图像，将其中的7000张图像划分为训练集，将剩余的3000张图像划分为测试集。
41.s130、建立改进yolox网络结构：在残差分支中额外引出特征，经过eca通道注意力后与另一分支的特征叠加，每个提取到的特征图都会有多个通道，各个通道上的特征通过叠加就可以得到完整的物体特征。
42.具体地，在本步骤中，一并参考图2，在yolox网络结构必不可少的骨干网络的各模块中，在残差分支中额外引出特征，经过eca通道注意力后与另一分支的特征叠加。每个提取到的特征图都会有多个通道，各个通道上的特征通过叠加就可以得到完整的物体特征。
43.进一步地，为了辨别各通道信息对目标将测的重要程度，嵌入eca通道注意力机制，对重要的特征分量所在通道赋予加大权重值，对非重要权重所在特征赋予较小权重，然后求加权和，eca模块结构如图3所示。作为一个示例，所述重要的特征分量的权重值范围为0.8～0.95，所述非重要的特征分量的权重值范围为0.1～0.25。
44.进一步地，在yolox网络结构中，浅层特征中物体位置信息丰富，而语义信息较少，深层特征中物体语义信息丰富，而位置信息较少，为了互补二者的特征信息，提出交叉融合策略应用于骨干网络中，如图4所示，对不同倍数下采样的特征实现融合
45.s140、利用所述训练集对所述改进yolox网络结构进行训练，得到实时路况目标检测模型。
46.s150、使用所述实时路况目标检测模型对实时路况目标进行检测。
47.本实施例的基于改进yolox的实时路况目标检测方法，在残差分支中额外引出特征，经过eca通道注意力后与另一分支的特征叠加，每个提取到的特征图都会有多个通道，各个通道上的特征通过叠加就可以得到完整的物体特征，可以有效提高实时路况目标检测精度和检测效率。在自制交通标志数据集上市进行测试，改进后的yolox算法在测试集上的检测精度和召回率都有明显提升。
48.本公开的另一方面，如图5所示，提供一种基于改进yolox的实时路况目标检测装置100，该装置100可以适用于前文记载的方法。所述装置100包括：
49.采集模块110，用于采集实时路况目标的数据集，并对数据集进行预处理；
50.划分模块120，用于按照预设的比例将预处理后的数据集划分为训练集和测试集；
51.建立模块130，用于建立改进yolox网络结构：在残差分支中额外引出特征，经过eca通道注意力后与另一分支的特征叠加，每个提取到的特征图都会有多个通道，各个通道上的特征通过叠加就可以得到完整的物体特征；
52.训练模块140，用于利用所述训练集对所述改进yolox网络结构进行训练，得到实时路况目标检测模型；
53.检测模块150，用于使用所述实时路况目标检测模型对实时路况目标进行检测。
54.本实施例的基于改进yolox的实时路况目标检测装置，在残差分支中额外引出特征，经过eca通道注意力后与另一分支的特征叠加，每个提取到的特征图都会有多个通道，各个通道上的特征通过叠加就可以得到完整的物体特征，可以有效提高实时路况目标检测
精度和检测效率。在自制交通标志数据集上市进行测试，改进后的yolox算法在测试集上的检测精度和召回率都有明显提升。
55.在一些实施方式中，所述eca通道注意力机制如下：
56.对重要的特征分量所在通道赋予加大权重值，对非重要的特征分量所在通道赋予较小权重，然后求加权和。
57.在一些实施方式中，所述重要的特征分量的权重值范围为0.8～0.95，所述非重要的特征分量的权重值范围为0.1～0.25。
58.在一些实施方式中，所述建立模块130，具体还用于：
59.yolox网络结构的各模块之间采用交叉融合策略，以使得浅层特征中物体特征信息和深层特征中物体特征信息互补。
60.本公开的另一方面，提供一种电子设备，包括：
61.一个或多个处理器；
62.存储单元，用于存储一个或多个程序，当所述一个或多个程序被所述一个或多个处理器执行时，能使得所述一个或多个处理器实现根据前文记载的所述的方法。
63.本公开的另一方面，提供一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时能实现根据前文记载的所述的方法。
64.其中，计算机可读介质可以是本公开的装置、设备、系统中所包含的，也可以是单独存在。
65.其中，计算机可读存储介质可是任何包含或存储程序的有形介质，其可以是电、磁、光、电磁、红外线、半导体的系统、装置、设备，更具体的例子包括但不限于：具有一个或多个导线的电连接、便携式计算机磁盘、硬盘、光纤、随机访问存储器(ram)、只读存储器(rom)、可擦式可编程只读存储器(eprom或闪存)、便携式紧凑磁盘只读存储器(cd-rom)、光存储器件、磁存储器件，或它们任意合适的组合。
66.其中，计算机可读存储介质也可包括在基带中或作为载波一部分传播的数据信号，其中承载了计算机可读的程序代码，其具体的例子包括但不限于电磁信号、光信号，或它们任意合适的组合。
67.可以理解的是，以上实施方式仅仅是为了说明本公开的原理而采用的示例性实施方式，然而本公开并不局限于此。对于本领域内的普通技术人员而言，在不脱离本公开的精神和实质的情况下，可以做出各种变型和改进，这些变型和改进也视为本公开的保护范围。

技术特征：
1.一种基于改进yolox的实时路况目标检测方法，其特征在于，所述方法包括：采集实时路况目标的数据集，并对数据集进行预处理；按照预设的比例将预处理后的数据集划分为训练集和测试集；建立改进yolox网络结构：在残差分支中额外引出特征，经过eca通道注意力后与另一分支的特征叠加，每个提取到的特征图都会有多个通道，各个通道上的特征通过叠加就可以得到完整的物体特征；利用所述训练集对所述改进yolox网络结构进行训练，得到实时路况目标检测模型；使用所述实时路况目标检测模型对实时路况目标进行检测。2.根据权利要求1所述的方法，其特征在于，所述eca通道注意力机制如下：对重要的特征分量所在通道赋予加大权重值，对非重要的特征分量所在通道赋予较小权重，然后求加权和。3.根据权利要求2所述的方法，其特征在于，所述重要的特征分量的权重值范围为0.8～0.95，所述非重要的特征分量的权重值范围为0.1～0.25。4.根据权利要求1至3任一项所述的方法，其特征在于，所述建立改进yolox网络结构，还包括：yolox网络结构的各模块之间采用交叉融合策略，以使得浅层特征中物体特征信息和深层特征中物体特征信息互补。5.一种基于改进yolox的实时路况目标检测装置，其特征在于，所述装置包括：采集模块，用于采集实时路况目标的数据集，并对数据集进行预处理；划分模块，用于按照预设的比例将预处理后的数据集划分为训练集和测试集；建立模块，用于建立改进yolox网络结构：在残差分支中额外引出特征，经过eca通道注意力后与另一分支的特征叠加，每个提取到的特征图都会有多个通道，各个通道上的特征通过叠加就可以得到完整的物体特征；训练模块，用于利用所述训练集对所述改进yolox网络结构进行训练，得到实时路况目标检测模型；检测模块，用于使用所述实时路况目标检测模型对实时路况目标进行检测。6.根据权利要求5所述的装置，其特征在于，所述eca通道注意力机制如下：对重要的特征分量所在通道赋予加大权重值，对非重要的特征分量所在通道赋予较小权重，然后求加权和。7.根据权利要求6所述的装置，其特征在于，所述重要的特征分量的权重值范围为0.8～0.95，所述非重要的特征分量的权重值范围为0.1～0.25。8.根据权利要求5至7任一项所述的装置，其特征在于，所述建立模块，具体还用于：yolox网络结构的各模块之间采用交叉融合策略，以使得浅层特征中物体特征信息和深层特征中物体特征信息互补。9.一种电子设备，其特征在于，包括：一个或多个处理器；存储单元，用于存储一个或多个程序，当所述一个或多个程序被所述一个或多个处理器执行时，能使得所述一个或多个处理器实现根据权利要求1至4任一项所述的方法。10.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序
被处理器执行时能实现根据权利要求1至4任一项所述的方法。

技术总结
本公开提供一种基于改进YOLOX的实时路况目标检测方法及装置。所述方法包括：采集实时路况目标的数据集，并对数据集进行预处理；按照预设的比例将预处理后的数据集划分为训练集和测试集；建立改进YOLOX网络结构：在残差分支中额外引出特征，经过ECA通道注意力后与另一分支的特征叠加，每个提取到的特征图都会有多个通道，各个通道上的特征通过叠加就可以得到完整的物体特征；利用所述训练集对所述改进YOLOX网络结构进行训练，得到实时路况目标检测模型；使用所述实时路况目标检测模型对实时路况目标进行检测。可以有效提高实时路况目标检测精度和检测效率。检测精度和检测效率。检测精度和检测效率。


技术研发人员：曹利蒲 杨继明 王军 陈婷婷 李涛 陈岩磊 张澈 王传鑫 田长风
受保护的技术使用者：北京华能新锐控制技术有限公司
技术研发日：2021.11.19
技术公布日：2022/3/7