计算机研究与发展

作者：侯鑫,曲国远,魏大洲,张佳程,

关键词：YOLOv3算法, 模型压缩, 迭代稀疏训练, 数据增强, 精度误差小,

随着无人机技术的成熟，配备摄像头的无人机被广泛应用于各个领域，自动高效地分析和理解从无人机收集的视觉数据非常重要.基于深度卷积神经网络的目标检测算法在许多实际应用上取得了惊人的成绩，但往往伴随着巨大的资源消耗和内存占用.因此，对于无人机上携带的计算能力受限的嵌入式设备来说，直接运行深度卷积神经网络非常具有挑战性.为了应对这些挑战，以经典的目标检测方法YOLOv3(youonlylookonce)为例，基于迭代稀疏训练的剪枝方式可以实现有效的模型压缩，同时通过组合不同数据增强方式与相关优化手段保证压缩前后检测器精度误差在可接受范围内.实验结果证明，基于迭代稀疏训练的剪枝方法在YOLOv3上取得了非常可观的压缩效果，并且将精度误差控制在了2%以内，为无人机目标检测实时应用提供了支持.

来源：2022年第4期

《计算机研究与发展》期刊编辑部

查看计算机研究与发展杂志2022年第4期