[发明专利]图像检测方法、装置、设备及存储介质

权利要求书

1.一种图像检测方法，其特征在于，包括：

获取待检测图像和模型训练数据集；

获取改进YOLO网络模型，其中，所述改进YOLO网络模型包含有自适应空间特征融合ASFF网络层、空间金字塔池化SPP网络层和融合因子α，所述融合因子α用于对所述SPP网络层处理后的特征图进行融合处理；

根据模型训练数据集训练所述改进YOLO网络模型，生成检测模型；

根据所述检测模型检测所述待检测图像，生成检测结果；

所述根据模型训练数据集训练所述改进YOLO网络模型，生成检测模型，包括：

基于所述YOLO网络模型包含的Darknet53网络层对所述模型训练数据集进行特征提取，生成三个层次的特征图；

基于所述SPP网络层分别对所述三个层次的特征图进行处理，生成三个输出特征图；

基于所述融合因子α和所述三个输出特征图，生成三个卷积特征图；

基于所述改进YOLO网络模型中的自适应空间特征融合ASFF网络层对所述三个卷积特征图进行处理；

将处理所述三个卷积特征图后得到的网络模型确定为检测模型；

所述基于所述融合因子α和所述三个输出特征图，生成三个卷积特征图，包括：

对第一输出特征图进行卷积处理，生成第一图像；

基于所述融合因子α分别对第一输出特征图和第二输出特征图，以及第二输出特征图和第三输出特征图进行融合处理，对应生成第二图像和第三图像；

分别对所述第一图像、所述第二图像和所述第三图像进行卷积处理，生成三个卷积特征图；

其中，所述第一输出特征图、所述第二输出特征图和所述第三输出特征图为所述三个输出特征图中的图像，且所述第一输出特征图、所述第二输出特征图和所述第三输出特征图的优先级依次从高到低。

2.根据权利要求1所述的方法，其特征在于，所述基于所述融合因子α分别对第一输出特征图和第二输出特征图，以及第二输出特征图和第三输出特征图进行融合处理，对应生成第二图像和第三图像，包括：

通过第一公式对第一输出特征图和第二输出特征图进行融合处理，生成第二图像；

通过所述第一公式对第二输出特征图和第三输出特征图进行融合处理，生成第三图像；

其中，所述第一公式含有所述融合因子α。

3.根据权利要求2所述的方法，其特征在于，所述第一公式还包括有输出特征图的输入通道和各输入通道对应的卷积核。

4.根据权利要求1所述的方法，其特征在于，所述基于所述改进YOLO网络模型中的自适应空间特征融合ASFF网络层对所述三个卷积特征图进行处理，包括：

在所述自适应空间特征融合ASFF网络层中，通过第二公式对所述三个卷积特征图进行特征融合处理；

其中，所述第二公式包括所述三个卷积特征图中每个卷积特征图上的特征点至第L个卷积特征图上对应特征点的特征向量，以及所述特征向量对应的权重参数，所述L的取值为1~3中的任意一个整数。