[发明专利]一种目标检测方法

权利要求书

1.一种目标检测方法，其特征在于，所述方法包括：

获取待检测图像；

利用目标检测模型对所述待检测图像进行目标检测处理，得到目标检测结果；所述目标检测模型是针对YOLOv4神经网络进行改进得到的模型；

所述目标检测模型包括：主干特征提取网络、颈部网络及预测结果生成网络；所述主干特征提取网络与所述颈部网络连接，所述颈部网络与所述预测结果生成网络连接；所述主干特征提取网络是基于双重注意力机制搭建的网络；所述颈部网络是使用基于双重注意力机制生成的Ghost模块搭建的网络。

2.如权利要求1所述的目标检测方法，其特征在于，所述获取待检测图像之前还包括：

基于双重注意力机制，搭建第一数量的跨阶段部分连接结构模块；

利用所述跨阶段部分连接结构模块，搭建所述主干特征提取网络；

基于双重注意力机制，生成第二数量的Ghost模块；

利用所述Ghost模块，搭建所述颈部网络；

利用第三数量的卷积模块，搭建所述预测结果生成网络；

依序建立所述主干特征提取网络、所述颈部网络及所述预测结果生成网络之间的连接关系，得到所述目标检测模型。

3.如权利要求2所述的目标检测方法，其特征在于，所述第一数量的跨阶段部分连接结构模块包括：第一跨阶段部分连接结构模块、第二跨阶段部分连接结构模块及第三跨阶段部分连接结构模块；

所述利用所述跨阶段部分连接结构模块，搭建所述主干特征提取网络，具体包括：

将所述第一跨阶段部分连接结构模块、所述第二跨阶段部分连接结构模块及所述第三跨阶段部分连接结构模块依序连接，得到所述主干特征提取网络；

其中，所述第一跨阶段部分连接结构模块用于接收目标图像，并针对所述目标图像进行卷积处理，提取所述目标图像的第一主干中间图像特征，以及，针对所述第一主干中间图像特征进行通道注意力加强处理和空间注意力加强处理，以得到所述目标图像的第一主干图像特征；

所述第二跨阶段部分连接结构模块用于接收所述第一主干图像特征，并针对所述第一主干图像特征进行卷积处理，以提取所述目标图像的第二主干中间图像特征，以及，针对所述第二主干中间图像特征进行通道注意力加强处理和空间注意力加强处理，以得到所述目标图像的第二主干图像特征；

所述第三跨阶段部分连接结构模块用于接收所述第二主干图像特征，并针对所述第二主干图像特征进行卷积处理，以提取所述目标图像的第三主干中间图像特征，以及，针对所述第三主干中间图像特征进行通道注意力加强处理和空间注意力加强处理，以得到所述目标图像的第三主干图像特征。

4.如权利要求2或3所述的目标检测方法，其特征在于，所述基于双重注意力机制，搭建第一数量的跨阶段部分连接结构模块，具体包括：

基于双重注意力机制，生成第一通道注意力子模块和第一空间注意力子模块；其中，所述第一通道注意力子模块用于对图像特征进行通道注意力加强处理；所述第一空间注意力子模块用于对图像特征进行空间注意力加强处理；

利用所述第一通道注意力子模块、第一空间注意力子模块及预设跨阶段部分连接结构子模块，生成各个跨阶段部分连接结构模块；其中，所述预设跨阶段部分连接结构子模块用于提取图像特征。