|
发表于 2019-1-21 17:15:58
|
显示全部楼层
你可以参考sample_nnie_main.c程序做修改
/mpp/sample/svp/multi-core/nnie/sample_nnie_main.c
0) RFCN(VI->VPSS->NNIE->VGS->VO).
1) Segnet(Read File).
2) FasterRcnnAlexnet(Read File).
3) FasterRcnnDoubleRoiPooling(Read File).
4) Cnn(Read File).
5) SSD(Read File).
6) Yolov1(Read File).
7) Yolov2(Read File).
8) LSTM(Read File).
9) Pvanet(Read File).
- /*GetResult*/
- /*if user has changed net struct, please make sure SAMPLE_SVP_NNIE_Rfcn_GetResult
- function's input datas are correct*/
- s32Ret = SAMPLE_SVP_NNIE_Rfcn_GetResult(pstParam,pstSwParam);
- SAMPLE_SVP_CHECK_EXPR_RET(HI_SUCCESS != s32Ret,s32Ret,SAMPLE_SVP_ERR_LEVEL_ERROR,
- "Error,SAMPLE_SVP_NNIE_Rfcn_GetResult failed!\n");
- /*draw result, this sample has 21 classes:
- class 0:background class 1:plane class 2:bicycle
- class 3:bird class 4:boat class 5:bottle
- class 6:bus class 7:car class 8:cat
- class 9:chair class10:cow class11:diningtable
- class 12:dog class13:horse class14:motorbike
- class 15:person class16:pottedplant class17:sheep
- class 18:sofa class19:train class20:tvmonitor*/
- s32Ret = SAMPLE_SVP_NNIE_RoiToRect(&(pstSwParam->stDstScore),
- &(pstSwParam->stDstRoi), &(pstSwParam->stClassRoiNum), pstSwParam->af32ScoreThr,HI_TRUE,&(pstSwParam->stRect),
- pstExtFrmInfo->stVFrame.u32Width, pstExtFrmInfo->stVFrame.u32Height,u32BaseWidth,u32BaseHeight);
- SAMPLE_SVP_CHECK_EXPR_RET(HI_SUCCESS != s32Ret,s32Ret,SAMPLE_SVP_ERR_LEVEL_ERROR,
- "Error(%#x),SAMPLE_SVP_NNIE_RoiToRect failed!\n",s32Ret);
- //Draw rect
- s32Ret = SAMPLE_COMM_SVP_NNIE_FillRect(&stBaseFrmInfo, &(pstSwParam->stRect), 0x0000FF00);
- SAMPLE_CHECK_EXPR_GOTO(HI_SUCCESS!=s32Ret, BASE_RELEASE,
- "SAMPLE_COMM_SVP_NNIE_FillRect failed, Error(%#x)!\n", s32Ret);
复制代码
|
|