[发明专利]一种基于海思平台的目标检测方法

权利要求书

1.一种基于海思平台的目标检测方法，其特征在于：包括如下具体步骤：

步骤1)、摄像头采集：对海思媒体处理平台(MPP)参数进行配置，包括使用HI_MPI_VB_Init、HI_MPI_SYS_Init等媒体接口配置并初始化视频缓存池和MPP系统，实现摄像头对场景信息的采集，实现视频输入、处理通道数据流的数据传递；

步骤2)、配置参数、加载模型：调用NNIE硬件单元，通过HI_MPI_SVP_NNIE_LoadModelAPI接口加载HI3559AV100芯片NNIE硬件单元支持的.wk目标检测模型文件，配置并初始化相应的模型参数，包含通过调用NNIE模块的基本接口获取卷积神经网络每段所需的辅助内存空间来计算获到总的内存空间大小、申请内存空间地址以及配置网络模型所需的参数；

步骤3)、模型前行推理：对摄像头获取的数据信息进行前向推理；首先通过HI_MPI_VPSS_GetChnFrame接口获得视频的帧图像，之后通过配置NNIE模块神经网络推理接口的对应参数，对获得的视频图像数据进行处理，得到图像经过推导后的结果，以及部分中间推导结果；

步骤4)、对步骤3)推理得到的结果进行进一步的处理，并通过输出接口输出到显示器上进行显示。

2.根据权利要求1所述的一种基于海思平台的目标检测方法，其特征在于，步骤2)所述通过HI_MPI_SVP_NNIE_LoadModelAPI接口加载的模型文件为在自建数据集上进行训练，并在训练完成后的模型上进行优化、转换得到的目标检测模型。

3.根据权利要求1所述的一种基于海思平台的目标检测方法，其特征在于，步骤4)所述方法，实现检测算法通过HDMI视频输出接口显示检测结果。