一种基于双模态融合的铜带表面缺陷检测方法

本发明涉及一种基于双模态融合的铜带表面缺陷检测方法，属铜带表面缺陷检测领域。包括：将近红外和可见光双模态铜带图片输入预先构建的双模态表面缺陷检测模型，再进行铜带图像缺陷检测，输出铜带缺陷的类型，分割掩码和置信度；双模态表面缺陷检测模型的构建包括：获取yolov8网络模型；利用NGAM注意力来优化yolov8网络模型的双主干网络，得到优化后的yolov8双主干网络；在优化后的yolov8双主干网络后添加近红外和可见光注意力融合模块NVAF得到构建好的双模态表面缺陷检测模型。本发明中NVAF模块和NGAM注意力结合在处理铜带表面缺陷检测任务时能够有效融合双模态信息，提高铜带表面缺陷检测的平均精度。