面向边缘端设备的轻量级金具检测方法OACSTPCD
为解决金具检测模型参数冗余大、终端部署检测实时性差等问题,文中设计一种基于网络通道剪枝的模型压缩算法。针对轻量级检测模型HRM-CenterNet存在的通道冗余问题,提出了分段稀疏化训练和全局通道剪枝方案。剪枝后的Pruned-HRMC模型参数量仅11.2 M,边缘端推理速度达到44 FPS,较原始模型体积缩小3倍,加速近2倍。结果表明,剪枝方案在金具检测任务上以极小的精度损失换取了大规模的模型压缩,极大满足了边缘端实时检测的工业需求。
苏丹;那琼澜;贺惠民;杨艺西;赵凯
国网冀北电力有限公司信息通信分公司,北京100053国网冀北电力有限公司信息通信分公司,北京100053国网冀北电力有限公司信息通信分公司,北京100053国家电网有限公司信息通信分公司,北京100761华北电力大学电子与通信工程系,河北保定071003
信息技术与安全科学
轻量级金具检测模型剪枝边缘端设备
《电力信息与通信技术》 2022 (11)
P.37-46,10
评论