海思部署：RuyiStudio流程解析(4)-前向传播SvpSampleCnnDetectionForword

本文链接：https://blog.youkuaiyun.com/weixin_40723264/article/details/130593680

该代码段展示了在嵌入式视觉平台SVP上进行神经网络前向传播的过程，使用HI_MPI_SVP_NNIE_Forward函数执行CNN网络预测，并通过HI_MPI_SVP_NNIE_Query函数查询任务是否完成。在前向传播中，输入和输出数据通过结构体数组处理，且可以设置是否产生中断。在查询状态时，会循环检测直到任务完成。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

前向传播 SvpSampleCnnDetectionForword(SVP_NNIE_ONE_SEG_DET_S *pstDetParam, SVP_NNIE_CFG_S *pstDetCfg)

1. 状态/变量声明(初始化)

HI_S32 SvpSampleCnnDetectionForword(SVP_NNIE_ONE_SEG_DET_S *pstDetParam, SVP_NNIE_CFG_S *pstDetCfg)

{

    HI_S32 s32Ret = HI_SUCCESS;//返回值,状态初始化

    SVP_NNIE_HANDLE SvpNnieHandle = 0;//

    SVP_NNIE_ID_E enNnieId = SVP_NNIE_ID_0;

    HI_BOOL bInstant = HI_TRUE;//bool

    HI_BOOL bFinish  = HI_FALSE;//bool

    HI_BOOL bBlock   = HI_TRUE;//bool

}

2. 网络前向传播

    //网络forward
    s32Ret = HI_MPI_SVP_NNIE_Forward(&SvpNnieHandle, pstDetParam->astSrc, &pstDetParam->stModel,
        pstDetParam->astDst, &pstDetParam->stCtrl, bInstant);
    CHECK_EXP_RET(HI_SUCCESS != s32Ret, s32Ret, "Error(%#x): CNN_Forward failed!", s32Ret);

（1）HI_MPI_SVP_NNIE_Forward()函数介绍

    HI_S32 HI_MPI_SVP_NNIE_Forward(SVP_NNIE_HANDLE *phSvpNnieHandle,
        const SVP_SRC_BLOB_S astSrc[], const SVP_NNIE_MODEL_S *pstModel,const SVP_DST_BLOB_S astDst[],
        const SVP_NNIE_FORWARD_CTRL_S *pstForwardCtrl,HI_BOOL bInstant);

该部分称为：多点输入输出的CNN网络预测。

phSvpNnieHandle: 指针，返回任务的ID

astSrc[]: 输入数据，类型为array数组

astDst[]:输出数据，为array类型

bInstant: 指示是否产生中断的标志。如果输出结果阻塞了下一次操作，请将bInstant设置为HI_TRUE。

支持的数据类型：

返回值：

在这个过程中会输出每层的信息，如下图所示：

3. Forward 状态查询

    //forward 状态查询
    cout<<"................... Query .................. "<<endl;
    s32Ret = HI_MPI_SVP_NNIE_Query(enNnieId, SvpNnieHandle, &bFinish, bBlock);
    while (HI_ERR_SVP_NNIE_QUERY_TIMEOUT == s32Ret)
    {
        USLEEP(100);
        s32Ret = HI_MPI_SVP_NNIE_Query(enNnieId, SvpNnieHandle, &bFinish, bBlock);
    }
    CHECK_EXP_RET(HI_SUCCESS != s32Ret, s32Ret, "Error(%#x): query failed!", s32Ret);

    return s32Ret;

（1）HI_MPI_SVP_NNIE_Forward()函数介绍

HI_S32 HI_MPI_SVP_NNIE_Query(SVP_NNIE_ID_E enNnieId,SVP_NNIE_HANDLE svpNnieHandle,HI_BOOL *pbFinish,HI_BOOL bBlock);
/*****************************************************************************
*   Prototype    : HI_MPI_SVP_NNIE_AddTskBuf
*   Description  : This API is used to record tskbuf in tskbuf array.
                   If the tskbuf has been recored, when user send the task with recored tskbuf,
                   the phyaddr will not be mapped again, this will improve the execution efficiency
*   Parameters   :
*                  const SVP_MEM_INFO_S* pstTskBuf    TskBuf
*   Return Value : HI_SUCCESS: Success;Error codes: Failure.
*   Spec         :
*   History:
*
*       1.  Date         : 2018-08-20
*          Author        :
*          Modification  : Created function
*
*****************************************************************************/

该部分作用：查询任务是否完成。