基于全卷积的无分支结构目标检测方法技术

本发明专利技术涉及一种基于全卷积的无分支结构标检测方法，包括以下步骤：步骤S1:获取目标检测数据集，并预处理，生成训练集；步骤S2:构建可重参数化的多分支全卷积网络模型；步骤S3:基于训练集对可重参数化的多分支全卷积网络模型进行训练，通过随机梯度下降方法，优化模型参数；步骤S4:对训练好的多分支全卷积网络模型进行重参数化，构建无分支全卷积网络，作为推理模型；步骤S5:将待测试图像输入推理模型进行测试。本发明专利技术能够有效地减少边缘小目标的漏检。的漏检。的漏检。

【技术实现步骤摘要】
基于全卷积的无分支结构目标检测方法


[0001]本专利技术涉及模式识别与计算机视觉领域，具体涉及一种基于全卷积的无分支结构目标检测方法。

技术介绍

[0002]目标检测是与计算机视觉和图像处理相关的计算机技术，是许多计算机视觉任务的基础，现阶段的目标检测算法为了解决网络退化问题设计更深的网络模型，网络中加入了旁路分支，拓扑结构越来越复杂。复杂的拓扑结构会导致的模型内存效率较低，因为每个分支的结果都需要保留到加法或级联之后，显著提高了内存占用。相比之下，无分支结构中的结果在操作完成时立即释放。并且复杂的拓扑结构的优点都是针对训练阶段，在应用时去除掉一些复杂的结构可以提高检测速度，重点关注于如何保证在测试阶段模型的精度。

技术实现思路

[0003]有鉴于此，本专利技术的目的在于提供一种基于全卷积的无分支结构目标检测方法，能够有效地减少边缘小目标的漏检，提高图像检测效率。
[0004]为实现上述目的，本专利技术采用如下技术方案：
[0005]一种基于全卷积的无分支结构目标检测方法，包括以下步骤：
[0006]步骤S1:获取目标检测数据集，并预处理，生成训练集；
[0007]步骤S2:构建可重参数化的多分支全卷积网络模型；
[0008]步骤S3:基于训练集对可重参数化的多分支全卷积网络模型进行训练，通过随机梯度下降方法，优化模型参数；
[0009]步骤S4:对训练好的多分支全卷积网络模型进行重参数化，构建无分支全卷积网络，作为推理模型；
[0010]步骤S5:将待测试图像输入推理模型进行测试。
[0011]进一步的，所述预处理具体为：对图像进行水平翻转和垂直翻转，图像翻转的概率设为0.5。
[0012]进一步的，所述步骤S2具体为:
[0013]步骤S21：使用ResNet作为主干网络的CenterNet目标检测模型作为基线模型,对ResNet的跳跃连接进行修改，一个ResNet基本块中包含两种跳跃连接，一种是当前层输出和输入特征图的尺度相同，直接对两个图像进行相加融合，一种是当前层输出特征图和输入特征图的尺度不同，使用1
×
1卷积对输入特征图进行通道变换后再与输出特征相加融合获得下一层的输入；
[0014]当尺度匹配时加入1
×
1卷积分支，不匹配时加入3
×
3卷积分支，并且每一个分支在卷积操作后加上批量归一化；对于尺度匹配的连接，设输入特征图为F1，通过当前层的3
×
3卷积操作后输出值为f1(F1)，设批量归一化操作为BN(X)，表示对输入X进行批量归一化，则BN(F1)表示对特征图F1进行批量归一化，则下一层的输入特征图Y1为：
[0015]Y1＝BN(g1(F1))+BN(f1(F1))+BN(F1)
[0016]其中g1(F1)是加入的1
×
1卷积分支的输出值，对于尺度匹不配的连接，设输入特征图为F2，通过当前层的3
×
3卷积操作后输出值为f2(F2)，则下一层的输入特征图Y2为：
[0017]Y2＝BN(g2(F2))+BN(f2(F2))+BN(H(F2))
[0018]其中g2(F2)是网络原来的1
×
1卷积分支的输出值，H(F2)是加入的3
×
3卷积分支的输出值；对每一个卷积操作加入分支结构，扩展为三个分支。
[0019]步骤S22：在步骤S21改进的ResNet网络后加上ECA注意力模块，将改进后的ResNet网络获得的特征值送入ECA注意力模块，构建可重参数化的多分支可重参数化的全卷积网络模型。
[0020]进一步的，所述步骤S3具体为：训练阶段使用随机初始化方法初始化网络参数，设置初始学习率设置为1.25e
‑
2，每过5轮将学习率乘以0.9998，使用动量为0.9的随机梯度下降优化方法，通过最小化损失获得模型的最优参数，得到训练好的可重参数化的多分支可重参数化的全卷积网络模型。
[0021]进一步的，所述步骤S4具体为:
[0022]步骤S41：将训练好的可重参数化的多分支可重参数化的全卷积网络模型中每一个卷积操作扩展为三个分支，若网络中一共包含n个卷积层，那么扩展后网络包含3n个处理过程，推理模型需要将3n个处理过程通过重参数化的方法转化为n个处理过程；用C
in
表示卷积操作中的参数输入通道数，C
out
表示输出通道数，对于尺度匹配的连接，设用表示当前层主干3
×
3卷积的权值，则为加入的旁路1
×
1卷积的权值，μ、σ、γ、β为批量归一化层的累积均值、标准差、学习的缩放因子和偏差，F为输出特征图,W为对应卷积层的权值；设为3
×
3卷积层之后批量归一化层的累积均值、标准差、学习的缩放因子和偏差；为加入的1
×
1卷积层之后批量归一化层的累积均值、标准差、学习的缩放因子和偏差，习的缩放因子和偏差，为输入值直接批量归一化层的累积均值、标准差、学习的缩放因子和偏差，那么训练阶段网络模型中对于输入特征图F，将使用Bn(F*W,μ,σ,γ,β)来表示批量归一化具体计算过程，*表示卷积：
[0023][0024][0025]步骤S42：根据步骤S41中批量归一化的计算公式，输入特征图F1与下一层的输入特征图Y1计算关系进一步表示为：
[0026][0027]根据训练好的模型权值重新构造测试模型需要的参数值，将多个分支结构的权值压缩到一个卷积中，将式子化简后的值使用一个卷积代替，简化后的计算公式为：
[0028][0029]测试阶段使用的压缩后的卷积的权值和偏移量{W'1,b'1}的公式如下:
[0030][0031][0032]得到测试阶段的为分支结构的计算关系为：
[0033]Y1＝F1*W'1+b'1[0034]步骤S43：根据步骤S42化简后的公式直接用于对于尺度不匹配的连接，输入通道数为C
in
，输出通道数为C
o枰t
的情况，用的情况，用表示当前层的主干3
×
3卷积的权值，的权值，为当前层1
×
1卷积的权值，则为加入的3
×
3卷积的权值；设为3
×
3卷积层之后批量归一化层的累积均值、标准差、学习的缩放因子和偏差；积均值、标准差、学习的缩放因子和偏差；为旁路1
×
1卷积层之后批量归一化层的累积均值、标准差、学习的缩放因子和偏差，μ
(3)
、σ
(3)
、γ
(3)
、β
(3)
为加入的旁路3
×
3卷积层之后批量归一化层的累积均值、标准差、学习的缩放因子和偏差，测试阶段使用的卷积的权值因此测试阶段使用的卷积的权值{W'2,b'2}的公式如下:
[0035][0036][00本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于全卷积的无分支结构目标检测方法，其特征在于，包括以下步骤：步骤S1:获取目标检测数据集，并预处理，生成训练集；步骤S2:构建可重参数化的多分支全卷积网络模型；步骤S3:基于训练集对可重参数化的多分支全卷积网络模型进行训练，通过随机梯度下降方法，优化模型参数；步骤S4:对训练好的多分支全卷积网络模型进行重参数化，构建无分支全卷积网络，作为推理模型；步骤S5:将待测试图像输入推理模型进行测试。2.根据权利要求1所述的基于全卷积的无分支结构目标检测方法，其特征在于，所述预处理具体为：对图像进行水平翻转和垂直翻转，图像翻转的概率设为0.5。3.根据权利要求1所述的基于全卷积的无分支结构目标检测方法，其特征在于，所述步骤S2具体为:步骤S21：使用ResNet作为主干网络的CenterNet目标检测模型作为基线模型，对ResNet的跳跃连接进行修改，一个ResNet基本块中包含两种跳跃连接，一种是当前层输出和输入特征图的尺度相同，直接对两个图像进行相加融合，一种是当前层输出特征图和输入特征图的尺度不同，使用1
×
1卷积对输入特征图进行通道变换后再与输出特征相加融合获得下一层的输入；当尺度匹配时加入1
×
1卷积分支，不匹配时加入3
×
3卷积分支，并且每一个分支在卷积操作后加上批量归一化；对于尺度匹配的连接，设输入特征图为F1，通过当前层的3
×
3卷积操作后输出值为f1(F1)，设批量归一化操作为BN(X)，表示对输入X进行批量归一化，则BN(F1)表示对特征图F1进行批量归一化，则下一层的输入特征图Y1为：Y1＝BN(g1(F1))+BN(f1(F1))+BN(F1)其中g1(F1)是加入的1
×
1卷积分支的输出值，对于尺度匹不配的连接，设输入特征图为F2，通过当前层的3
×
3卷积操作后输出值为f2(F2)，则下一层的输入特征图Y2为：Y2＝BN(g2(F2))+BN(f2(F2))+BN(H(F2))其中g2(F2)是网络原来的1
×
1卷积分支的输出值，H(F2)是加入的3
×
3卷积分支的输出值；对每一个卷积操作加入分支结构，扩展为三个分支。步骤S22：在步骤S21改进的ResNet网络后加上ECA注意力模块，将改进后的ResNet网络获得的特征值送入ECA注意力模块，构建可重参数化的多分支可重参数化的全卷积网络模型。4.根据权利要求1所述的基于全卷积的无分支结构目标检测方法，其特征在于，所述步骤S3具体为：训练阶段使用随机初始化方法初始化网络参数，设置初始学习率设置为1.25e
‑
2，每过5轮将学习率乘以0.9998，使用动量为0.9的随机梯度下降优化方法，通过最小化损失获得模型的最优参数，得到训练好的可重参数化的多分支可重参数化的全卷积网络模型。5.根据权利要求1所述的基于全卷积的无分支结构目标检测方法，其特征在于，所述步骤S4具体为：步骤S41：训练好的可重参数化的多分支可重参数化的全卷积网络模型中每一个卷积操作包含三个分支，若网络中一共包含n个卷积层，那么网络包含3n个处理过程，推理模型
需要将3n个处理过程通过重参数化的方法转化为n个处理过程；用C
in
表示卷积操作中的参数输入...

【专利技术属性】
技术研发人员：柯逍，林心茹，刘浩，
申请(专利权)人：福州大学，
类型：发明
国别省市：