面向空基边缘平台轻量化目标智能检测

作者：王文庆; 丰林; 刘洋; 马晓华; 杨东方

来源：西安邮电大学学报, 2021, 26(03): 61-72.

DOI：10.13682/j.issn.2095-6533.2021.03.010

• 通道剪枝
• 目标检测
• 深度学习
• 模型压缩
• 稀疏训练

针对基于深度学习的目标检测模型面向空基边缘平台计算开销和参数量较大等问题,提出一种面向空基边缘平台的轻量化目标智能检测方法。以YOLOv4为基础,在训练过程中对卷积层缩放因子加入L1正则约束,增强通道的稀疏性。对所有的卷积核权重进行排序,给定权重阈值,去除小于该阈值的卷积核,通过微调得到剪枝后的精简YOLOv4模型。实验结果表明,精简YOLOv4模型比初始YOLOv4模型大小减少了97%,计算量减少了73%,边缘平台帧率提升了一倍,具有更快的运算速度以及更低的计算功耗。

• 单位
  西安邮电大学; 自动化学院