一种基于知识蒸馏和剪枝技术的光伏板缺陷识别方法

本发明涉及图像识别技术领域,具体是公开一种基于知识蒸馏和剪枝技术的光伏板缺陷识别方法，步骤建立缺陷图像数据和标注图像数据，将标注图像数据划分为训练集、验证集、测试集用于SRGAN模型训练、验证和测试，得到超分辨图像数据，将步标注图像数据和超分辨图像数据划分为训练集、验证集、测试集，分别训练、验证和测试教师模型和学生模型得到的性能最优权重，得到蒸馏训练后的学生模型，得到约束训练后的学生模型，得到剪枝后的学生模型，对剪枝后的学生模型进行回调训练。该方法可以降低耗时、昂贵的人工检测成本等，可提高产能，提高实际工业应用价值，具备高检测精度和轻量化优点，更适于光伏板图像缺陷识别。