利索能及
我要发布
收藏
专利号: 2021113927966
申请人: 江苏科技大学
专利类型:发明专利
专利状态:已下证
更新日期:2026-03-02
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种水下图像语义分割模型,其特征在于,包括:预处理模块,用于对水下图像进行预处理得到输入图像;

编码器,包括五个阶段编码模块,第四阶段编码模块包含卷积注意力模块,所述卷积注意力模块中的通道注意力子模块对通道设置不同的权重,所述卷积注意力模块中的空间注意力子模块对空间信息设置不同的权重;所述编码器用于按权重进行特征提取;

解码器,用于恢复输入图像的细节信息和空间位置信息,得到分割之后的图像。

2.根据权利要求1所述的水下图像语义分割模型,其特征在于,在编码器中,所述第四阶段编码模块输出的特征图经过通道注意力子模块,分别进行平均池化和最大池化后再进行相加和激活操作,得到特征图E17;所述特征图E17经过空间注意力子模块,沿通道分别对维度计算均值和最大值后再进行相加、卷积和激活操作。

3.根据权利要求1所述的水下图像语义分割模型,其特征在于,所述四个阶段编码模块中,第一阶段和第二阶段编码模块包含两个卷积层,第三阶段、第四阶段和第五阶段编码模块包含三个卷积层。

4.根据权利要求1所述的水下图像语义分割模型,其特征在于,所述解码器包含四个阶段解码模块,对编码后的特征图进行上采样、拼接、卷积和激活操作;其中第一、二、三阶段解码模块进行两次卷积,第四阶段解码模块进行三次卷积。

5.根据权利要求4所述的水下图像语义分割模型,其特征在于,所述第一阶段解码模块中将第一阶段上采样后的特征图与所述卷积注意力模块输出的特征图进行拼接;第二阶段解码模块中将第二阶段上采样后的特征图与所述第三阶段编码模块输出的特征图进行拼接;第三阶段解码模块中将第三阶段上采样后的特征图与所述二阶段编码模块输出的特征图进行拼接;第四阶段解码模块中为第四阶段上采样后的特征图与所述一阶段编码模块输出的特征图进行拼接。

6.根据权利要求1所述的水下图像语义分割模型,其特征在于,所述通道注意力子模块采用ELU激活函数。

7.根据权利要求1所述的水下图像语义分割模型,其特征在于,所述预处理模块对水下图像进行像进行尺寸调整并转化为RGB三通道,得到输入图像。

8.一种水下图像语义分割方法,其特征在于,包括如下步骤:(1)对水下图像进行预处理;

(2)将预处理后的图像输入语义分割模型;所述语义分割模型使用UNet网络,在所述语义分割模型的编码器中包含卷积注意力模块,所述卷积注意力模块包含通道注意力子模块和空间注意力子模块,通道注意力子模块对通道设置不同的权重,空间注意力子模块对空间信息设置不同的权重;所述编码器按照权重进行特征提取;(3)采用预处理的数据集对所述语义分割模型进行训练;

(4)将待分割的水下图像输入训练好的语义分割模型,输出分割后的图像。

9.根据权利要求8所述的水下图像语义分割方法,其特征在于,步骤(3)中采用预处理的数据集对语义分割模型进行训练,利用损失函数对训练模型进行约束,然后反向传播对参数进行迭代优化。