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

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