(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202211508063.9 (22)申请日 2022.11.29 (71)申请人 昆明理工大 学 地址 650000 云南省昆明市一 二一大街文 昌路68号 (72)发明人 杨晶晶　张强　王镭　邓桂方　 蒋一凡　 (74)专利代理 机构 北京壹川鸣知识产权代理事 务所(特殊普通 合伙) 11765 专利代理师 陈春风 (51)Int.Cl. G06T 5/00(2006.01) G06T 7/11(2017.01) G06T 7/136(2017.01) G06V 10/44(2022.01)G06V 10/80(2022.01) G06N 3/08(2006.01) G06N 3/04(2006.01) (54)发明名称 地砖图像阴影去除方法、 装置、 计算机设备 及存储介质 (57)摘要 本发明涉及计算机技术领域， 特别是涉及一 种地砖图像阴影去除方法、 装置、 计算机设备及 存储介质， 所述地砖图像阴影去除方法包括： 获 取地砖图像， 所述地砖图像包括地砖阴影图像以 及地砖无阴影图像； 将所述地砖阴影图像进行阈 值分割得到地砖阴影图像的mask掩码； 构建基于 空间自适应归一化与特征融合的处理模型以及 权重损失函数； 使用所述无阴影图像以及所述 mask掩码对所述处理模型进行训练直至所述权 重损失函数满足要求； 使用训练好的处理模型处 理目标地砖图像以去除目标地砖图像中的阴影。 本发明结合了空间自适应归一化与特征融合， 具 有良好的特征提取能力， 更有针对性地对图像中 的阴影区域进行处理， 适用于实际施工环境下的 地砖阴影去除。 权利要求书3页 说明书7页 附图4页 CN 115546073 A 2022.12.30 CN 115546073 A 1.地砖图像阴影去除方法， 其特 征在于， 所述 地砖图像阴影去除方法包括： 获取地砖图像， 所述 地砖图像包括 地砖阴影图像以及地 砖无阴影图像； 将所述地砖阴影图像进行阈值分割得到地 砖阴影图像的mask掩码； 构建基于空间自适应归一 化与特征融合的处 理模型以及权 重损失函数； 使用所述无阴影图像以及所述mask掩码对所述处理模型进行训练直至所述权重损失 函数满足要求； 使用训练好的处 理模型处 理目标地 砖图像以去除目标地 砖图像中的阴影； 所述基于空间自适应归一化与特征融合的处理模型包括若干卷积层、 若干个反卷积层 以及若干个 基于掩码的空间自适应归一 化模块； 若干个卷积层顺次连接； 若干个反卷积层顺次连接后设置 于顺次连接的若干个卷积层之后； 每个所述基于掩码的空间自适应归一 化模块均设置 两个相邻的反卷积层之间； 所述权重损失函数 由图像特征相似度函数 、 图像亮度相似性函数 以及图 像对比度相似性 函数 构成。 2.根据权利要求1所述的地砖图像阴影去除方法， 其特征在于， 所述将所述地砖阴影图 像进行阈值分割得到地 砖阴影图像的mask掩码， 包括： 将地砖阴影图像的像素进行分类， 其中阴影区域标记为白色， 非阴影区域标记为黑色， 得到地砖阴影图像的mask掩码。 3.根据权利要求1所述的地砖图像阴影去除方法， 其特征在于， 所述基于掩码的空间自 适应归一 化模块包括 一个第一卷积层以及两个第二卷积层； mask掩码输入第一卷积层， 第一卷积层的输出分别作为两个第二卷积层的输入， 由两 个第二卷积层分别得到参数γ和参数β； 参数γ和参数β 与BatchNorm层的输出х进行运算， 得到   作为下一层 的输入。 4.根据权利要求1所述的地砖图像阴影去除方法， 其特征在于， 第 一卷积层与第 一个反 卷积层之间通过 连接函数Co ncat建立特 征通道融合。 5.根据权利要求1所述的地砖图像阴影去除方法， 其特征在于， 所述卷积层 设置有6个， 所述反卷积层设置有6个， 所述基于掩码的空间自适应归一化模块设置有4个， 且第四个卷 积层与第二个 基于掩码的空间自适应归一 化模块间建立特 征值相加融合。 6.根据权利要求1所述的地砖图像阴影去除方法， 其特征在于， 所述权重损失函数 具体为： 所述图像特 征相似度函数 具体为：权　利　要　求　书 1/3 页 2 CN 115546073 A 2其中： 表示第i张标签图像的特征向量， 表示第i张网络结果图的特征向量， n为图 像的数量； 所述图像亮度相似性 函数 具体为： 其中： 表示图像χ的平均亮度， 即 ， 为第i张图像χ的亮度， 表示 图像y的平均亮度， 即 ， 为第i张图像y的亮度； 为防止分母系数为零的常 数； 所述图像对比度相似性 函数 具体为： 其中： 表示图像χ 的像素标准差， 即 ， 表示图像у的 像素标准差， 即 ， 是防止分母为 零的常数。 7.地砖图像阴影去除装置， 其特 征在于， 所述 地砖图像阴影去除装置包括： 地砖图像获取模块， 用于地砖图像的获取， 所述地砖图像包括地砖阴影图像以及地砖 无阴影图像； 分割模块， 用于将所述 地砖阴影图像进行阈值分割得到地 砖阴影图像的mask掩码； 处理模块， 用于构建基于空间自适应归一化与特征融合的处理模型以及权重损失函 数； 训练模块， 用于使用所述无阴影图像以及所述mask掩码对所述处理模型进行训练直至 所述权重损失函数满足要求； 识别模块， 用于使用训练好的处理模型处理目标地砖图像以去除目标地砖图像中的阴 影； 所述基于空间自适应归一化与特征融合的处理模型包括若干卷积层、 若干个反卷积层 以及若干个 基于掩码的空间自适应归一 化模块； 若干个卷积层顺次连接；权　利　要　求　书 2/3 页 3 CN 115546073 A 3