基于微调式结构化剪枝的架空输电线路绝缘子检测算法

作者：梁爽^*; 刘丹丹; 季堂煜

来源：电力信息与通信技术, 2023, 21(11): 1-6.

DOI：10.16543/j.2095-641x.electric.power.ict.2023.11.01

• 深度学习
• 架空输电线路绝缘子
• 结构化剪枝
• 知识蒸馏

针对现行架空输电线路绝缘子检测算法存在的复杂度高和实时性差等问题，文章提出基于结构化剪枝的检测算法。首先引入HardSwish激活函数，加速模型收敛；其次，使用K-means++算法对锚框聚类，使锚框更贴合目标；在训练过程中利用归一化层缩放因子γ进行L1正则化稀疏训练，优化滤波器以及卷积核权重，剪除信息量较少的通道。对于剪枝后的模型，通过知识蒸馏进行微调，恢复精度。采用DIOU-NMS算法进行后处理，优化模型实际检测结果。实验表明：文章模型浮点数运算量减小至8.2GFLOPs，有效降低了模型的复杂度；m AP0.5高达96.8%，性能优于其他模型；检测速度可达21.23fps，可以较好地满足实时性检测需求。

• 单位
  上海电力大学