基于NVIDIA TX2的行人检测方法

摘要

本发明涉及一种基于NVIDIA TX2的行人检测方法，包括下列步骤：行人数据集的建立；选取YOLOv3‑Tiny网络，进行如下调整：增大网络预设的缩放分辨率，以便于检测小目标；将网络设计为仅使用特定宽高比的anchor，以使得网络更加适应行人的检测；引入MobileNetv2，对于YOLOv3‑Tiny网络中的卷积层在执行卷积操作时，将通道和空间区域分开考虑，将卷积层进行拆分；利用BN层的缩放因子，来衡量通道的重要性，实现通道裁剪，模型减小；将通道裁剪后的模型移植到开发板，并对行人检测过程进行内存优化，采用统一内存寻址方案，使得在CPU填充数据之后，GPU能够直接访问。