本发明专利技术属于计算机视觉技术领域，具体涉及一种基于Laplace特征约束的双分支非对称实时语义分割算法，包括以下步骤：步骤10：根据待识别的图像M，搜索相关数据并制作成数据集S；步骤20：对数据集S中的图像进行预处理，包括图像的翻转、旋转、缩放和裁剪；步骤30：搭建神经网络模型框架，包括图像处理模块、特征提取模块、ASSP模块、特征融合模块、Laplace二阶微分特征提取模块、特征细化提取模块和卷积块注意力模块步骤40：训练神经网络模型；步骤50：将待识别的图像M输入到训练好的神经网络模型中，得到识别结果；本发明专利技术能够提高图像推理速度的同时减少延迟并提高分割的准确率。减少延迟并提高分割的准确率。减少延迟并提高分割的准确率。
【技术实现步骤摘要】
一种基于Laplace特征约束的双分支非对称实时语义分割算法
[0001]本专利技术属于计算机视觉
，具体涉及一种基于Laplace特征约束的双分支非对称实时语义分割算法。
技术介绍
[0002]图像语义分割是计算机视觉领域中的一个重要部分，其任务是将图像中每个像素精确地分类到它所属的类别，并与图像本身的视觉呈现内容相匹配。因此，图像语义分割也被称为像素级的图像分类任务
[0003]常见的语义分割网络在设计过程中通常不考虑推理速度和计算成本，而更注重网络的骨干结构设计，如ResNet
‑
101、Xception和HRNet等。这些结构多层次、复杂而深入，能够从更大范围的空间中提取综合特征，具备更强的特征表达能力。然而，这些庞大的主干模型往往计算复杂度高、推理速度较低，这对于一些实际应用来说不太合适，这些应用需要低延迟及竞争力较好的预测精度。因此，如何在高效的推理速度及较高的分割准确率之间寻求平衡，已成为一项具有挑战性的任务。
技术实现思路
[0004]本专利技术的目的是：旨在提供一种基于Laplace特征约束的双分支非对称实时语义分割算法，能够主要通过特征融合模块和特征细化提取模块降低图像计算的复杂度，进而提高推理速度，同时还具有较好的较好的分割准确率，进而使该模型具有分割准确率的同时提高图像的推理速度。
[0005]为实现上述技术目的，本专利技术采用的技术方案如下：
[0006]一种基于Laplace特征约束的双分支非对称实时语义分割算法，包括以下步骤：
[0007]步骤10：根据待识别的图像M，搜索相关数据并制作成数据集S；
[0008]步骤20：对数据集S中的图像进行预处理，包括图像的翻转、旋转、缩放和裁剪；
[0009]步骤30：搭建神经网络模型框架，包括图像处理模块、特征提取模块、ASSP模块、特征融合模块、Laplace二阶微分特征提取模块、特征细化提取模块和卷积块注意力模块
[0010]步骤40：训练神经网络模型；
[0011]步骤50：将待识别的图像M输入到训练好的神经网络模型中，得到识别结果。
[0012]所述特征提取模块使用MobilenetV3网络，将第5个步长为2的下采样bneck模块替换为步长为1、卷积核大小为5
×
5的bneck模块。
[0013]所述ASSP模块参照Deeplabv3网络中ASSP模块设计，将其中三个空洞卷积的扩张率更换为2、7、15。
[0014]所述特征融合模块首先将输入进行2倍上采样，之后与特征提取模块中对应分辨率的低级特征图进行通道相连，将得到的256通道的特征图通过通道自注意力机制获得256个权重值，将权重值与与256通道的特征图对应相乘，获得输出特征图。
[0015]所述通道自注意力机制首先将输入特征图分别进行全局平局池化与全局最大池化，分别获得256个尺寸为1
×
1的特征，之后将两组特征分别先后经过第一次1
×
1卷积、ReLu激活函数和第二次1
×
1卷积，第一个1
×
1卷积输入特征通道数为256，输出特征通道数为64，第二个1
×
1卷积输入特征通道数为64，输出特征通道数为256，然后将两组特征相加融合，最后经过sigmoid函数激活后与输入特征相乘。
[0016]所述Laplace二阶微分特征提取模块使用Laplace卷积核对输入图像的3通道分别卷积得到三通道特征图，将得到的三通道特征取绝对值后相加，得到二阶微分特征。
[0017]所述特征细化提取模块具有3层，卷积核的个数分别为16，64，128，Laplace二阶微分特征提取模块得到的特征图先通过16个7
×
7的卷积核获得16个特征图，16个特征图再分别通过4个5
×
5的卷积核获得64个特征图，64个特征图分别通过2个3
×
3的卷积核获得128个特征图。
[0018]所述卷积块注意力模块包括通道自注意力机制和空间自注意力机制。
[0019]所述通道自注意力机制首先将输入特征图分别进行全局平局池化与全局最大池化，分别获得128个尺寸为1
×
1的特征，之后将两组特征分别先后经过第一次1
×
1卷积、ReLu激活函数和第二次1
×
1卷积，第一个1
×
1卷积输入特征通道数为128，输出特征通道数为32，第二个1
×
1卷积输入特征通道数为32，输出特征通道数为128，然后将两组特征相加融合，最后经过sigmoid函数激活后与输入特征相乘。
[0020]所述空间自注意力机制首先将输入特征图在通道维度分别进行全局平局池化与全局最大池化，将获得的两个特征图在通道维度相连，之后通过7
×
7的卷积获得一个空间约束特征。
[0021]本专利技术能够主要通过特征融合模块和特征细化提取模块降低图像计算的复杂度，进而提高推理速度，同时还具有较好的较好的分割准确率，进而使该模型具有分割准确率的同时提高图像的推理速度。
附图说明
[0022]本专利技术可以通过附图给出的非限定性实施例进一步说明。
[0023]图1为本专利技术一种基于Laplace特征约束的双分支非对称实时语义分割算法的总体结构示意图；
[0024]图2为本专利技术一种基于Laplace特征约束的双分支非对称实时语义分割算法的特征融合模块示意图；
[0025]图3为本专利技术一种基于Laplace特征约束的双分支非对称实时语义分割算法的特征细化提取模块示意图；
[0026]图4为本专利技术一种基于Laplace特征约束的双分支非对称实时语义分割算法的卷积块注意力模块示意图；
[0027]图5为本专利技术一种基于Laplace特征约束的双分支非对称实时语义分割算法与其他算法的对比示意图。
具体实施方式
[0028]为了使本领域的技术人员可以更好地理解本专利技术，下面结合附图和实施例对本发
明技术方案进一步说明。
[0029]如图1
‑
5所示，本专利技术的一种基于双分支非对称结构的语义分割算法，
[0030]首先搜素与待识别图像M相关的数据集S，然后建立基于双分支非对称结构神经网络架构，将图像进行图像处理模块、特征提取模块、ASSP模块、特征融合模块、Laplace二阶微分特征提取模块、特征细化提取模块和卷积块注意力模块；
[0031]通过搭建的神经网络模型框架训练神经网络模型包括：
[0032]1、使用MobilenetV3网络，将第5个步长为2的下采样bneck模块替换为步长为1、卷积核大小为5
×
5的bneck模块去将数据进行特征提取得特征图。
[0033]2、参照Deeplabv3网络中ASSP模块设计对数据进行处理，将其中三个空洞卷积的扩张率更换本文档来自技高网...
【技术保护点】
【技术特征摘要】
1.一种基于Laplace特征约束的双分支非对称实时语义分割算法，其特征在于，包括以下步骤：步骤10：根据待识别的图像M，搜索相关数据并制作成数据集S；步骤20：对数据集S中的图像进行预处理，包括图像的翻转、旋转、缩放和裁剪；步骤30：搭建神经网络模型框架，包括图像处理模块、特征提取模块、ASSP模块、特征融合模块、Laplace二阶微分特征提取模块、特征细化提取模块和卷积块注意力模块；步骤40：训练神经网络模型；步骤50：将待识别的图像M输入到训练好的神经网络模型中，得到识别结果。2.根据权利要求1所述的一种基于Laplace特征约束的双分支非对称实时语义分割算法，其特征在于：所述特征提取模块使用MobilenetV3网络，将第5个步长为2的下采样bneck模块替换为步长为1、卷积核大小为5
×
5的bneck模块。3.根据权利要求1所述的一种基于Laplace特征约束的双分支非对称实时语义分割算法，其特征在于：所述ASSP模块参照Deeplabv3网络中ASSP模块设计，将其中三个空洞卷积的扩张率更换为2、7、15。4.根据权利要求1所述的一种基于Laplace特征约束的双分支非对称实时语义分割算法，其特征在于：所述特征融合模块首先将输入进行2倍上采样，之后与特征提取模块中对应分辨率的低级特征图进行通道相连，将得到的256通道的特征图通过通道自注意力机制获得256个权重值，将权重值与与256通道的特征图对应相乘，获得输出特征图。5.根据权利要求4所述的一种基于Laplace特征约束的双分支非对称实时语义分割算法，其特征在于：所述通道自注意力机制首先将输入特征图分别进行全局平局池化与全局最大池化，分别获得256个尺寸为1
×
1的特征，之后将两组特征分别先后经过第一次1
×
1卷积、ReLu激活函数和第二次1
×
1卷积，第一个1
×
1卷积输入特征通道数为256，输出特征通道数为64，第二个1
×
1卷积输入特征通道数为64，输出特征通道数为256，然后将两组特征相加融合，最...
【专利技术属性】
技术研发人员：武文昊，祝志芳，王志远，卢全国，王红州，刘源杰，贺一博，罗平平，何玉灵，李志豪，
申请(专利权)人：南昌工程学院，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。