一种基于开集分类的二阶段变电站设备视觉缺陷检测方法

本发明公开了一种基于开集分类的二阶段变电站设备视觉缺陷检测方法，通过构建数据集；构建改进YOLOv8n网络模型；训练改进YOLOv8n网络模型；构建Swin Transformer为主干网络的开集分类网络；训练开集分类网络；测试所提出的二阶段目标检测算法的效果。本发明提供的基于开集分类的二阶段变电站设备视觉缺陷检测方法，针对变电站缺陷检测中目标类别多样、环境复杂并且存在大量的视觉干扰，提出了一种新型的二阶段目标检测框架，采用一阶段目标检测算法和开集分类算法相结合，实现了更强的抗干扰能力，在变电站缺陷检测中表现出优异的性能。