基于嵌入式Jetson TX2的高原鼠兔目标检测  

Target detection of Ochotona curzoniae based on embedded Jetson TX2

作　　者：陈海燕[1] 贾明明 赵文力 王婵飞[1] CHEN Haiyan;JIA Mingming;ZHAO Wenli;WANG Chanfei(School of Computer and Communication,Lanzhou University of Technology,Lanzhou Gansu 730050,China)

机构地区：[1]兰州理工大学计算机与通信学院,兰州730050

出　　处：《计算机应用》2023年第1期98-103,共6页journal of Computer Applications

基　　金：国家自然科学基金资助项目(62161019,62061024)。

摘　　要：高原鼠兔目标检测是对其进行种群数量统计及种群动态变化研究的基础,但传统的高原鼠兔智能监测系统的目标检测硬件设备大,在抽样采集数据时移动性较弱。针对此问题,提出一种可部署到便携式设备Jetson TX2上的基于改进YOLOv3模型的目标检测方法。该方法将YOLOv3的主干网络DarkNet53替换成MobileNet,并利用剪枝、微调等方法构建轻量级高原鼠兔目标检测模型,再将轻量化模型部署到Jetson TX2上。自然场景下高原鼠兔目标检测实验的结果表明:所提方法的检测平均精度(AP)、每秒检测帧数(FPS)和模型大小分别为97.36%、36和14.88 MB,优于主干网络替换后未裁剪的YOLOv3模型及原始YOLOv3模型,相较于原YOLOv3模型,AP在仅下降1.05个百分点的情况下,FPS提升了620%,模型大小压缩了93.67%,能够部署在便携设备上进行实时且准确的高原鼠兔目标检测。Target detection of Ochotona curzoniae is the basis for its population statistics and population dynamic changes research, but the traditional intelligent monitoring systems of Ochotona curzoniae has a large target detection hardware equipment and weak mobility in sampling and collecting data. To solve the above problems, an improved YOLOv3-based target detection method that can be deployed to the portable device Jetson TX2 was proposed. The lightweight Ochotona curzoniae target detection model was constructed by replacing Darknet53, backbone network of YOLOv3, with MobileNet and using pruning and fine-tuning methods to further reduce the model size. Next, the model was deployed on the portable target detection device Jetson TX2. Experimental results of Ochotona curzoniae target detection in natural scenes show that the proposed method has the detection Average Precision(AP), detection Frames per Second(FPS) and model size of 97. 36%, 36 and 14. 88 MB, respectively, which are better than those of the improved YOLOv3 model without pruning and the original YOLOv3 model;with the AP only reduced by 1. 05 percentage points, the proposed method has the detection speed improved by 620% and the model size compressed by 93. 67%, which can be deployed to portable devices for real-time and accurate detection of Ochotona curzoniae.

关 键 词：目标检测 YOLOv3 轻量化 模型剪枝 Jetson TX2 高原鼠兔 

分 类 号：TP391.4[自动化与计算机技术—计算机应用技术]