留言板

尊敬的读者、作者、审稿人, 关于本刊的投稿、审稿、编辑和出版的任何问题, 您可以本页添加留言。我们将尽快给您答复。谢谢您的支持!

姓名
邮箱
手机号码
标题
留言内容
验证码

基于宇航级NPU的轻量化Yolov5算法的目标检测系统

刘冰 周海 卞春江 成晓蕾 王鹏飞 张彪

刘冰, 周海, 卞春江, 成晓蕾, 王鹏飞, 张彪. 基于宇航级NPU的轻量化Yolov5算法的目标检测系统[J]. 空间科学学报. doi: 10.11728/cjss2024-0103
引用本文: 刘冰, 周海, 卞春江, 成晓蕾, 王鹏飞, 张彪. 基于宇航级NPU的轻量化Yolov5算法的目标检测系统[J]. 空间科学学报. doi: 10.11728/cjss2024-0103
Lightweight Yolov5 algorithm target detection system based on space-grade NPU[J]. Chinese Journal of Space Science. doi: 10.11728/cjss2024-0103
Citation: Lightweight Yolov5 algorithm target detection system based on space-grade NPU[J]. Chinese Journal of Space Science. doi: 10.11728/cjss2024-0103

基于宇航级NPU的轻量化Yolov5算法的目标检测系统

doi: 10.11728/cjss2024-0103 cstr: 32142.14.cjss2024-0103
基金项目: 中国科学院青年创新促进会基金项目

Lightweight Yolov5 algorithm target detection system based on space-grade NPU

  • 摘要: 针对基于深度学习的目标检测算法的网络结构复杂且所需计算量过大导致其不易部署在资源十分有限的星上图像处理平台这一问题,本文提出了一种基于宇航级神经网络处理器(Neural-network Processing Unit,NPU)的卷积神经网络加速设计,利用改进后的Yolov5s网络在NPU上实现快速图像处理功能。该设计将网络中与NPU适配度低的结构替换掉,并引入注意力机制来弥补轻量化网络带来的精度损失,得到优化后的网络后通过公开数据集VOC在GPU上进行迭代训练,经CPU-NPU并行协同处理设计后,实现将图像处理的三部分并行执行,充分利用Yulong810A平台的有限计算和存储资源。实验证明,优化后的网络不仅参数量减小82%,精度较原版Yolov5s网络也有所提升,mAP值达82.35%。该算法部署到Yulong810A星上处理平台后的目标检测速度达47.67fps/s,比原Yolov5s网络的速度提升两倍以上,实现了一个更轻量化、更快速的目标检测系统。

     

  • 加载中
计量
  • 文章访问数:  253
  • HTML全文浏览量:  22
  • PDF下载量:  10
  • 被引次数: 

    0(来源:Crossref)

    0(来源:其他)

出版历程
  • 收稿日期:  2024-08-20
  • 录用日期:  2024-11-18
  • 修回日期:  2024-10-29
  • 网络出版日期:  2024-12-31

目录

    /

    返回文章
    返回