电极头故障的检测方法、装置、设备以及存储介质

本申请公开了一种电极头故障的检测方法、装置、设备以及存储介质，涉及人工智能技术领域，该方法包括：构建深度学习网络，所述深度学习网络包括主干网络、头部网络、特征增强模块和注意力机制模块；对电极头图像数据进行预处理，所述预处理方式包括归一化、尺寸调整和数据增强；将预处理后的电极头图像数据输入所述深度学习网络进行特征提取得到特征；通过特征增强模块对提取的特征进行增强处理；利用注意力机制模块对增强后的特征进行筛选和整合；将筛选和整合后的特征输入头部网络进行分类和定位，输出电极头故障检测结果；在电极头故障检测结果中检测到故障时，输出报警信息。