一种基于海思平台的小目标检测方法

摘要

本发明公开了一种基于海思平台的小目标检测方法。首先配置媒体处理平台参数驱动摄像头，采集视频数据，实现数据传递，量化并加载目标检测算法模型，初始化目标检测算法和神经网络推理计算单元；然后使用视频处理子系统模块对接受到的图像队列进行裁剪和缩放得到新的图像队列，根据新的图像队列的图像分配内存，并使用专用的神经网络推理模块对图像队列并行进行前行推理；其次对得到的推理结果进行处理；最后通过HDMI接口输出显示。本发明通过使用海思HI3559AV100芯片内部的NNIE模块和自带的高效媒体处理功能，使整个检测过程在一块嵌入式ARM板子上进行，对比现有的方法，实现了对小目标检测的性能要求，更好的发挥了深度学习的优势。