-
公开(公告)号:CN117495808A
公开(公告)日:2024-02-02
申请号:CN202311461883.1
申请日:2023-11-06
Applicant: 江南大学
IPC: G06T7/00 , G06N3/0464 , G06N3/08
Abstract: 本发明提供基于自适应空间注意机制的金属表面小目标缺陷检测方法,其在不增加计算资源的前提下,可以针对金属表面小缺陷目标进行更好地分类,确保训练保留全局信息的同时更加注重于输入特征的重要信息,在不降低推理速度的基础上实现提升模型的识别精度。其将自适应空间注意机制嵌入到现有的网络模型中,在自适应空间注意机制中,对输出特征进行自适应池化操作;将自适应池化操作输出的特征图沿通道进行拼接操作得到最具代表性而又兼顾上下文信息的特征图,然后利用空洞空间注意机制在增大感受野的同时,提升缺陷特征空间上的依赖关系;最后,通过通道拆分合并的操作对特征图进行降维可以更好地保留所提取的特征而又降低了计算资源。