一种适合无人机部署的轻量级YOLO剪枝模型

• 作者: 王睿旸 施欣妤 陈伟 陆科名 陈曦珑
• 来源期刊: 电脑编程技巧与维护
• 关键词: 目标检测 稀疏训练 通道剪枝 知识蒸馏
• 年，卷（期）: 2021,（1）
• 所属期刊栏目: 人工智能与应用
• 页码范围: 133-136,176
• 页数: 5页
• 语种: 中文

摘要

计算复杂度一直制约着目标检测算法在边缘端设备中的部署,利用模型剪枝方法,对流行的目前检测算法YOLOv3进行了精简,提出了一种适合于无人机部署的目标检测模型,在几乎不降低模型精度的前提下大大降低了模型的参数量和浮点计算量.通过L1正则化、几何中心匹配、通道剪枝、层剪枝、知识蒸馏等方法,对YOLOv3算法进行了剪枝和优化.