一种基于视觉神经网络的圆形衬套缺陷检测方法及系统

本发明涉及计算机视觉技术领域，尤其是涉及一种基于视觉神经网络的圆形衬套缺陷检测方法及系统。所述方法，包括：获取训练用图像数据集，并对图像数据集进行预处理；基于获取的图像数据集进行图像分割网络搭建，包括利用线性投影将图像分割为不重叠的块；利用编码器提取分割后的图像特征注意力关系，并通过线性分类器对图像分割网络的输出进行分类；根据图像分割网络的输出分类进行缺陷检测网络搭建，对缺陷检测网络中的每个阶段引入通道注意力机制，并将所有注意力信息反馈至原特征图；利用缺陷检测网络进行检测结果的输出，本发明通过添加通道注意力机制可以帮助网络对图像中重要的目标信息进行突出，提高网络对目标部分的感知能力。