首页 | 本学科首页   官方微博 | 高级检索  
     

基于嵌入式平台与优化YOLOv3 的航拍目标检测方法
引用本文:郭智超.基于嵌入式平台与优化YOLOv3 的航拍目标检测方法[J].兵工自动化,2022,41(3):10-15,20.
作者姓名:郭智超
作者单位:海军航空大学,山东 烟台 264001
基金项目:国家自然科学基金项目(51605487)
摘    要:针对部署在嵌入式平台的目标检测模型在检测航拍目标时存在的检测速率低、耗时高、存储容量低的问 题,提出一种基于优化YOLOv3 算法的航拍目标检测方法。通过模型剪枝极大地减少了模型参数量,使用二分 K-means 对传统的锚框聚类算法进行优化改进,引入CIOU 损失函数加强边界框回归效果,再经TensorRT 对模型优 化加速后将该检测模型部署到JetsonTX2 平台上。选取大量不同类别不同环境的航拍图像制作数据集进行实验对比。 结果表明:优化后的算法在检验不同航拍图像目标时平均精度可达到83.9%,对每张图片的检测速度从2.8 FPS 提升 至14.7 FPS,满足精确性和实时性要求。

关 键 词:目标检测  YOLOv3算法  神经网络  深度学习  JetsonTX2平台
收稿时间:2021/11/3 0:00:00
修稿时间:2021/12/28 0:00:00

Aerial Target Detection Method Based on Embedded Platform and Optimized YOLOv3
Guo Zhichao,Xu Junming,Liu Aidong.Aerial Target Detection Method Based on Embedded Platform and Optimized YOLOv3[J].Ordnance Industry Automation,2022,41(3):10-15,20.
Authors:Guo Zhichao  Xu Junming  Liu Aidong
Abstract:Aiming at the problems of low detection rate, high time consumption and low storage capacity of target detection model deployed on embedded platform when detecting aerial targets, proposes an aerial target detection method based on optimized YOLOv3 algorithm. The number of model parameters is greatly reduced by model pruning. The traditional anchor box clustering algorithm is optimized and improved by using binary K-means. The CIOU loss function is introduced to enhance the effect of bounding box regression. After the model is optimized and accelerated by TensorRT, the detection model is deployed on JetsonTX2 platform. By selecting a large number of aerial images of different types and different environments to make data sets, the experimental results show that the average accuracy of the optimized algorithm can reach 83. 9% when detecting targets in different aerial images, and the detection speed of each image is improved from 2. 8 FPS to 14. 7 FPS, which meets the requirements of accuracy and real-time.
Keywords:target detection  YOLOv3 algorithm  neural network  deep learning  JetsonTX2 platform
本文献已被 万方数据 等数据库收录!
点击此处可从《兵工自动化》浏览原始摘要信息
点击此处可从《兵工自动化》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号