计算机研究与发展2022,Vol.59Issue(4) :882-893.DOI:10.7544/issn1000-1239.20200986

基于迭代稀疏训练的轻量化无人机目标检测算法

A Lightweight UAV Object Detection Algorithm Based on Iterative Sparse Training

侯鑫 曲国远 魏大洲 张佳程
计算机研究与发展2022,Vol.59Issue(4) :882-893.DOI:10.7544/issn1000-1239.20200986

基于迭代稀疏训练的轻量化无人机目标检测算法

A Lightweight UAV Object Detection Algorithm Based on Iterative Sparse Training

侯鑫 1曲国远 2魏大洲 2张佳程3
扫码查看

作者信息

  • 1. 中国科学院计算技术研究所 北京 100190
  • 2. 中国航空无线电电子研究所 上海 200241
  • 3. 北京邮电大学信息与通信工程学院 北京 100876
  • 折叠

摘要

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

关键词

YOLOv3算法/模型压缩/迭代稀疏训练/数据增强/精度误差小

引用本文复制引用

基金项目

国家重点研发计划(2018YFC0809300)

国家重点研发计划(2107YFB0202105)

国家重点研发计划(2016YFB0200803)

国家重点研发计划(2017YFB0202302)

国家自然科学基金(61972376)

北京市自然科学基金(L182053)

出版年

2022
计算机研究与发展
中国科学院计算技术研究所 中国计算机学会

计算机研究与发展

CSTPCDCSCD北大核心
影响因子:2.649
ISSN:1000-1239
参考文献量48
段落导航相关论文