[发明专利]一种基于分类网路的电力复合绝缘子憎水性分类系统

权利要求书

1.一种基于分类网路的电力复合绝缘子憎水性分类系统，其特征在于，由图像采集装置、数据传输链路、训练服务器、TX2端侧推理设备、显示模块组成：

所述图像采集装置用于拍摄不同天气、不同地形条件下的输电线路上带水珠的复合绝缘子图像；所述训练服务器用于数据增强、数据集存储、训练RepVGG分类模型、模型转换；所述数据传输链路用于向TX2端侧推理设备传输图像数据；所述TX2端侧推理设备用于模型移植和数据推理；所述显示模块用于对接收到的分类结果进行显示。

2.根据权利要求1所述的一种基于分类网路的电力复合绝缘子憎水性分类系统，其特征在于，所述数据增强包含颜色变换和几何变换，颜色变换由高斯噪声、高斯模糊、图像擦除、图像填充组成；几何变换由图像翻转，图像旋转，图像随机裁剪，图像随机缩放组成。

3.根据权利要求1所述的一种基于分类网路的电力复合绝缘子憎水性分类系统，其特征在于，所述数据集存储，是将图像大小为640×640，图像通道为RGB的绝缘子带水珠的图片，按照HC1-HC3、HC4-HC5、HC6-HC7三个尺度，来标记划分憎水性等级，按照4：1比例划分训练集和测试集，并存放于不同文件夹下。

4.根据权利要求1所述的一种基于分类网路的电力复合绝缘子憎水性分类系统，其特征在于，所述数据传输链路使用MQTT作为通信协议。

5.根据权利要求1所述的一种基于分类网路的电力复合绝缘子憎水性分类系统，其特征在于，所述模型转换技术路线为：首先将model.pth转换为model.onnx，再将model.onnx转换为model.trt。

6.根据权利要求1所述的一种基于分类网路的电力复合绝缘子憎水性分类系统，其特征在于，所述训练RepVGG分类模型，在进行模型在训练时，为RepVGG网络中每一个3x3卷积层添加平行的1x1卷积分支和恒等映射分支，构成一个RepVGG Block。

7.根据权利要求1所述的一种基于分类网路的电力复合绝缘子憎水性分类系统，其特征在于，所述模型推理，是训练RepVGG模型完成后，对模型做等价转换，实现方式为：y＝x+g(x)+f(x)转换为y＝h(x)，其中，

x表示每层的输入，f(x)表示3×3卷积；

g(x)是一个1×1的卷积，h(x)恒等映射分支的输出。

8.根据权利要求1所述的一种基于分类网路的电力复合绝缘子憎水性分类系统，其特征在于，所述模型移植为TRT量化后的模型，实现方法为：

x_Q＝clamp(0,N_levels-1,x_int)

其中，x_int表示浮点型值x映射为整型值，x表示输入的浮点型变量，Δ表示量化的步长，z表示零点，x_Q表示量化后的定点值，N_levels＝256，clamp用于限定量化取值范围。