面向边缘端设备的轻量级金具检测方法

作     者：苏丹 那琼澜 贺惠民 杨艺西 赵凯 SU Dan;NA Qionglan;HE Huimin;YANG Yixi;ZHAO Kai

作者机构：国网冀北电力有限公司信息通信分公司北京100053 国家电网有限公司信息通信分公司北京100761 华北电力大学电子与通信工程系河北保定071003 

出 版 物：《电力信息与通信技术》 (Electric Power Information and Communication Technology)

年 卷 期：2022年第20卷第11期

页      面：37-46页

学科分类：08[工学] 0804[工学-仪器科学与技术] 

基　　金：国网冀北电力有限公司科技项目“基于边缘智能的信息通信机房智能人员管控技术研究”(B3018E210001)。 

主　　题：轻量级 金具检测 模型剪枝 边缘端设备 

摘      要：为解决金具检测模型参数冗余大、终端部署检测实时性差等问题,文中设计一种基于网络通道剪枝的模型压缩算法。针对轻量级检测模型HRM-CenterNet存在的通道冗余问题,提出了分段稀疏化训练和全局通道剪枝方案。剪枝后的Pruned-HRMC模型参数量仅11.2 M,边缘端推理速度达到44 FPS,较原始模型体积缩小3倍,加速近2倍。结果表明,剪枝方案在金具检测任务上以极小的精度损失换取了大规模的模型压缩,极大满足了边缘端实时检测的工业需求。