(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210635875.3 (22)申请日 2022.06.07 (71)申请人 重庆大学 地址 400044 重庆市沙坪坝区沙坪坝正 街 174号 申请人 重庆小康工业 集团股份有限公司 　 武汉招商 滚装运输有限公司重庆分 公司 (72)发明人 杨波　肖猛　王时龙　张正萍　 唐小丽　徐佳　 (74)专利代理 机构 重庆航图知识产权代理事务 所(普通合伙) 50247 专利代理师 胡小龙 (51)Int.Cl. G06T 7/00(2017.01)G06N 3/08(2006.01) G06N 3/04(2006.01) G06V 10/44(2022.01) G06V 10/82(2022.01) G01N 21/88(2006.01) (54)发明名称 精细坐标注意力模块及其在表面缺陷检测 中的应用 (57)摘要 本发明公开了一种精细坐标注 意力模块， 包 括坐标信息聚合单元、 跨维度交互单元和注意力 生成单元； 所述坐标信息聚合单元用于对中间特 征图分别在高度和宽度方向上编码坐标信息， 并 在高度和宽度方向上分别进行坐标平均池化和 坐标最大池化处理， 生成四组聚合特征； 所述跨 维度交互单元采用一个带降维层的卷积块， 使所 述坐标信息聚合单元生成的四组聚合特征共用 该卷积块并独立进行特征变换， 得到四组变换特 征； 所述注 意力生成单元利用注 意力权重将四组 变换特征分别生成在高度和宽度方向上的两组 坐标注意力， 利用该两组坐标注 意力对中间特征 图进行重校准， 得到输出特征图。 本发明还公开 了一种如上所述精细坐标注意力模块在表面缺 陷检测中的应用。 权利要求书2页 说明书12页 附图4页 CN 114972280 A 2022.08.30 CN 114972280 A 1.一种精细坐标注意力模块， 其特征在于： 包括坐标信 息聚合单元、 跨维度交互单元和 注意力生成单 元； 所述坐标信 息聚合单元用于对中间特征图分别在高度和宽度方向上编码坐标信 息， 并 在高度和宽度方向上分别进行坐标平均池化和坐标最大池化处 理， 生成四组聚合特 征； 所述跨维度交互单元采用一个带降维层的卷积块， 使所述坐标信 息聚合单元生成的四 组聚合特 征共用该 卷积块并独立进行 特征变换， 得到四组变换 特征； 所述注意力生成单元利用注意力权重将四组变换特征分别生成在高度和宽度方向上 的两组坐标注意力， 利用该两组坐标注意力对中间特 征图进行重校准， 得到 输出特征图。 2.根据权利要求1所述的精细坐标注意力模块， 其特 征在于： 对于中间特征图 其中， xk表示第k个通道输入的特征 图； 表示三维张量； H表示高度； W表示宽度； C表示 通道； 所述坐标信息聚合单 元在高度方向上编码坐标信息得到的两组聚合特 征为： 其中， 分别表示在高度方向上编码坐标信息得到的两组聚合特征； 和 分别表示基于高度方向的坐标平均池化和坐标最大池化； 所述坐标信息聚合单 元在宽度方向上编码坐标信息得到的两组聚合特 征为： 其中， 分别表示在宽度方向上编码坐标信息得到的两组聚合特征； 和 分别表示基于 宽度方向的坐标平均池化和坐标最大池化。 3.根据权利要求2所述的精细坐标注意力模块， 其特征在于： 所述跨维度交互单元可以 表示为： 其中， 表示在高度方向上的两 组变换特征； 表示在宽度 方向上的两 组变换特征； f1×1(V0,V1)表示变换运算函数， 实现了对聚合特征的跨维度交互； 和 均为1×1卷积核； r 表示维度降低率； δ( ·)表示ReLU函数。 4.根据权利要求3所述的精细坐标注意力模块， 其特征在于： 所述注意力生成单元中， 高度方向和宽度方向上的注意力权 重为： 其中， 和 分别表示在高度和宽度方向上的两组坐标注意力； σ权　利　要　求　书 1/2 页 2 CN 114972280 A 2(·)表示Sigmo id函数； 利用两组坐标注意力对中间特 征图X进行重校准， 得到： Z＝fco(fco(X,Yh),Yw) 其中， fco(·,·)表示坐标级别的乘法， Z表示中间特征图X经过重校准后得到的输出特 征图。 5.一种如权利要求1 ‑4任一项所述精细坐标注意力模块在表面缺陷检测中的应用， 其 特征在于： 利用卷积神经网络进行表面缺陷检测， 将所述精细坐标注意力模块嵌入到所述 卷积神经网络内嵌入内。权　利　要　求　书 2/2 页 3 CN 114972280 A 3