摘要
本发明公开基于改进YOLOv7的绝缘子缺陷检测方法,将待识别图片输入预先训练获得的网络模型,输出初步预测框;利用SIoU‑NMS改进非极大值抑制算法对待识别图片输出的初步预测框进行过滤,获得最终预测框,从而获得待识别图片的缺陷类型。预先训练获得网络模型:预处理图片样本的目标真实框,获得多个固定尺寸的锚框;网络模型将图片样本划分成多个网格;基于固定尺寸的锚框,以每个网格为中心生成多个不同尺寸的预测框;利用总损失函数分别计算图片样本的目标真实框和不同尺寸的预测框之间的差距,迭代更新网络模型的网络参数;若总损失函数收敛于定值,则停止迭代,输出最终的网络模型。