一种基于RT-DETR算法的特征蒸馏树线故障识别方法

本发明公开了一种基于RT‑DETR算法的特征蒸馏树线故障识别方法，属于电力防灾技术领域，包括构建树线故障图像数据集，划分为对抗图像数据和标注图像数据；利用对抗图像数据训练得到超分辨率图像数据；将超分辨率图像数据与标注图像数据混合，划分为训练集、验证集和测试集；分别对教师模型和学生模型进行训练、验证和测试，获得性能最优权重；利用性能最优权重，得到特征蒸馏后的学生模型；利用蒸馏后的学生模型得到剪枝后的学生模型；然后进行蒸馏微调，获得树线故障识别模型；将树线故障识别模型部署至边缘端系统。与现有技术相比，本发明能够降低人工检测成本，提升故障检测的实时性，减少对云端服务器的依赖，更适用于树线故障识别场景。