Input
热力图
Conv+UpSample+Element Add
F2
特征金字塔网络
Concat
缺陷定位结果
自注意力模块
多尺度特征 F1
Conv(Convolution):卷积层CBR(Conv+BatchNorm+ReIu):卷积+批归一化+ReluMP(Max-Pooling):最大池化CA(Channel Attention):通道注意力UpSample:上采样Element Add:对应元素和
F3
原始图像检测结果
自注意力特征F4
卷积神经网络模型
CBR+MP+CA
目标定位模块
数据扩充
注意力机制
Conv
SENet
底层特征PF1
初始目标检测图
随机拼接
Fr
目标筛选及热力图生成网络
图像增广
掩码交叉注意力模块
轮廓细化模块
图像增强