基于网络剪枝的轻量化道路损坏检测模型研究

• 作者: 范良辰 夏颖慧 李雨诗 陈绪君
• 来源期刊: 信息技术
• 学科: 工学
• 关键词: 道路损坏检测 YOLOv4算法 网络剪枝 模型压缩 模型加速
• 年，卷（期）: 2022,（3）
• 所属期刊栏目: 研究与探讨|RESEARCH AND DISCUSSION
• 页码范围: 96-102,108
• 页数: 8页
• 分类号: TP301.5
• 语种: 中文
• DOI: 10.13274/j.cnki.hdzj.2022.03.017

摘要

为了解决目标检测模型参数冗余大、终端部署检测实时性差等问题,文中引入一种基于网络通道剪枝的模型压缩算法.针对轻量级YOLOv4-MobileNet检测模型存在的通道冗余问题,提出了被动式剪枝和主动式剪枝两种优化方案.剪枝后模型Pruned_Model参数量仅5.9M,推理速度达到324.8FPS,较原始模型压缩过5倍,加速近3倍,总体精度损失仅2.1％.结果表明,剪枝方案在道路损坏检测任务上以极小的精度损失换取了大规模的模型压缩.