关键词:
目标检测
深度学习
嵌入式系统
实时性
Jeston TX1
摘要:
目标检测作为装甲车辆环视态势感知系统的关键技术,可有效提高装甲车辆内乘员的观测能力和其应对潜在威胁的能力。基于深度学习的目标检测算法在配备高性能显卡的服务器上可以达到实时性、准确性的要求,然而面对装甲车的战场应用环境,配备高性能显卡服务器和使用网络远程连接服务器的解决方案均不具备现实可行性。因此,本文研究基于嵌入式平台的深度学习实时目标检测,解决基于深度卷积神经网络的目标检测算法在嵌入式平台无法满足实时性要求的问题,提出了一种轻量级的卷积神经网络MTYOLO,并将其移植到Jeston TX1嵌入式平台,实现了基于嵌入式平台的深度学习实时目标检测,具有积极的现实意义。首先,本文研究了基于深度学习的目标检测算法,对基于分类的目标检测算法和基于回归的目标检测算法进行了研究,旨在找到相对适合嵌入式平台的目标检测算法。实验结果证明,基于回归的目标检测算法更适合本文的应用场景,因此选择该类算法作为本文的基础方法。其次,本文研究了面向嵌入式平台的高效卷积神经网络,从减少模型体积和模型所需计算量两个角度出发,研究一种有效的网络模型压缩与加速的方法。实验结果证明,MobileNet将SSD模型压缩为原模型体积的1/5的同时,检测速度提升了 43%。因此,深度可分离卷积对模型压缩与加速有着更优秀的效果,更加适合本课题的应用场景。再次,在Tiny Yolov2的基础上,本文设计了一种计算能力需求小、目标检测效果稳定的轻量级卷积神经网络MTYOLO。通过引入深度可分离卷积、瓶颈层、多层特征融合以及特征金字塔的方法,改善了目标检测的效果。通过在多种数据集上训练并测试,验证了 MTYOLO目标检测算法的可行性。实验结果表明,在KITTI数据集上,MTYOLO目标检测精度可达到60.2mAP,检测速度可达到153FPS(Frame Per Second)。最后,本文针对Jeston TX1设计了车载智能硬件终端,给出了硬件的系统架构以及终端的实现方案。将MTYOLO目标检测算法移植到该平台上,并使用模型压缩方法对模型进一步优化。通过整体系统功耗实验和检测性能评估,验证了车载智能硬件终端的可靠性、稳定性以及MTYOLO目标检测算法的可行性。