3516a 尝试跑通GMM 模型之二

HI_VOID GMMSampleSingle()
{
#if 1
	IVE_SRC_IMAGE_S stIveImg;
	IVE_DST_IMAGE_S stIveFg;
	IVE_DST_IMAGE_S stIveBg;


	HI_BOOL bInstant = HI_TRUE;   // must be true

	
	IVE_HANDLE hIveHandle;        // handle
	IVE_MEM_INFO_S *pstDst;       //gmm model
        IVE_GMM_CTRL_S stGMMCtrl;     //stGmmCtrl
	IplImage *pimgGray=NULL;
	IplImage *pfg=NULL;
	IplImage *pfgRGB=NULL;
	IplImage *pBgGray=NULL;
	HI_S32 s32Result;             // if s32Result is 0 success
	HI_U32   FrameNum = 1;

	CvSize S;
	
	 string buff = "list1.txt";
	 const char * outputDir = "output";

	 ifstream infile;

	infile.open(buff.c_str());
	getline(infile, buff);

	IplImage * pimg=cvLoadImage(buff.c_str());
	HI_U16 u16Width=pimg->width;
	HI_U16 u16Height=pimg->height;

       cout<<"u16Width "<<u16Width<<endl;
       cout<<"u16Height "<<u16Height<<endl;

	if(NULL == pimgGray)
		{
			printf("coming into pimgGray\n");
			S = cvSize(pimg->width,pimg->height);
			pimgGray = cvCreateImage(S,8,1);
			if(HI_NULL == pimgGray)
			{
				printf("pimgGray creat error");
				goto FAILURE;
			}
		}

        
        // pstDst
	//pstDst = (IVE_DST_MEM_INFO_S *)malloc(sizeof(IVE_DST_MEM_INFO_S));
	//HI_CHECK_ET_GOTO(pstDst,HI_NULL,FAILURE);
      


	stGMMCtrl.u0q16InitWeight = 3276; //0.05
    	stGMMCtrl.u0q16BgRatio = 52428;   //0.8
    	stGMMCtrl.u22q10MaxVar = (2000 << 10);
    	stGMMCtrl.u22q10MinVar = (200 << 10);
    	stGMMCtrl.u22q10NoiseVar = (225 << 10);
    	stGMMCtrl.u8q8VarThr = 1600;
    	stGMMCtrl.u8ModelNum =3;
    	stGMMCtrl.u0q16LearnRate = 327;
         
        // init 
	memset(&stIveImg, 0, sizeof(IVE_SRC_IMAGE_S));
	memset(&stIveFg, 0,sizeof(IVE_DST_IMAGE_S));
	memset(&stIveBg, 0, sizeof(IVE_DST_IMAGE_S));
	
	//init 
	SAMPLE_COMM_IVE_CheckIveMpiInit();


	while(getline(infile, buff))
	{
		 pimg=cvLoadImage(buff.c_str());
		if (!pimg)
		{
			cout<<"error input image"<<endl;
			break;
		}
		cvCvtColor(pimg,pimgGray,CV_BGR2GRAY);
		cout<<"pimg->nChannels(): "<<pimg->nChannels<<endl;
	        cout<<"pimgGray->nChannels():   "<<pimgGray->nChannels<<endl;

		// source image
		if(HI_NULL == stIveImg.pu8VirAddr[0])
		{
		s32Result = SAMPLE_COMM_IVE_CreateImage(&stIveImg,IVE_IMAGE_TYPE_U8C1,u16Width,u16Height);
		cout<<"s32Result stIveImg HI_CreateIveImage "<<s32Result<<endl;   // if s32Result is 0 success	
		}
		cout<<stIveImg.u32PhyAddr[0]<<" "<<"stIveImg.u32PhyAddr[0]"<<endl;
 		memcpy(stIveImg.pu8VirAddr[0],pimgGray->imageData,pimgGray->width * pimgGray->height);
		
        	

		//foreground
		if(HI_NULL == stIveFg.pu8VirAddr[0])
		{
		s32Result = SAMPLE_COMM_IVE_CreateImage(&stIveFg,IVE_IMAGE_TYPE_U8C1, u16Width,u16Height);
		cout<<"s32Result Fg HI_CreateIveImage "<<s32Result<<endl;

		}
		cout<<stIveFg.u32PhyAddr[0]<<" "<<"stIveFg.u32PhyAddr[0]"<<endl;
		memcpy(stIveFg.pu8VirAddr[0],pimgGray->imageData, pimgGray->width * pimgGray->height);


		// background
		if(HI_NULL == stIveBg.pu8VirAddr[0])
		{
		s32Result = SAMPLE_COMM_IVE_CreateImage(&stIveBg,IVE_IMAGE_TYPE_U8C1, u16Width,u16Height);
		cout<<"s32Result Bg HI_CreateIveImage "<<s32Result<<endl;
		//HI_CHECK_ET_GOTO(s32Result,HI_FAILURE,FAILURE);
		}
		cout<<stIveBg.u32PhyAddr[0]<<" "<<"stIveBg.u32PhyAddr[0]"<<endl;
		memcpy(stIveBg.pu8VirAddr[0],pimgGray->imageData, pimgGray->width * pimgGray->height);


		//pstModel
		
		pstDst->u32Size = stGMMCtrl.u8ModelNum * 7 * pimg->width * pimg->height;
		s32Result  = SAMPLE_COMM_IVE_CreateMemInfo(pstDst, pstDst->u32Size);
		if (s32Result!= HI_SUCCESS)
		 {
		   printf("SAMPLE_COMM_IVE_CreateMemInfo fail\n");
		 }
                memset(pstDst->pu8VirAddr, 0, pstDst->u32Size);
		

			
		time_t start_t,end_t;
		start_t = clock();
		//cout<<"will get into GMM"<<endl;

cout<<stIveImg.u32PhyAddr[0]<<" "<<"stIveImg.u32PhyAddr[0]"<<endl;
cout<<stIveFg.u32PhyAddr[0]<<" "<<"stIveFg.u32PhyAddr[0]"<<endl;
cout<<stIveBg.u32PhyAddr[0]<<" "<<"stIveBg.u32PhyAddr[0]"<<endl;
cout<<pstDst->u32PhyAddr<<" "<<"pstDst->u32PhyAddr"<<endl; 

s32Result = HI_MPI_IVE_GMM(&hIveHandle,&stIveImg,&stIveFg,&stIveBg,pstDst,&stGMMCtrl,bInstant);
 if (HI_SUCCESS != s32Result)
    {
       printf("s32Result (%#x)\n", s32Result);
    }
		end_t = clock();
		long cost_t = (long)(end_t - start_t);
		cout<<"cost_t: "<<cost_t<<endl;
		if(HI_SUCCESS != s32Result)
		{
			goto FAILURE;
		}

		if(pfg == HI_NULL)
		{
			S = cvSize(pimg->width,pimg->height);
			pfg = cvCreateImage(S,8,1);
			if(HI_NULL == pfg)
			{
				goto FAILURE;
			}
		}

		if(HI_NULL == pfgRGB)
		{
			S = cvSize(pimg->width,pimg->height);
			pfgRGB = cvCreateImage(S,8,3);
			if(HI_NULL == pfgRGB)
			{
				goto FAILURE;
			}
		}
		if(HI_NULL == pBgGray)
		{
			S = cvSize(pimg->width,pimg->height);
			pBgGray = cvCreateImage(S,8,1);
			if(HI_NULL == pBgGray)
			{
				goto FAILURE;
			}
		}
		memcpy(pfg->imageData, stIveFg.pu8VirAddr[0],pimg->width * pimg->height);
		memcpy(pBgGray->imageData,stIveBg.pu8VirAddr[0],pimg->height * stIveBg.u16Stride[0]);
		cvCvtColor( pfg, pfgRGB, CV_GRAY2BGR );

		FrameNum++;
		
	}

FAILURE:
	 IVE_MMZ_FREE(stIveImg.u32PhyAddr[0],stIveImg.pu8VirAddr[0]);
         IVE_MMZ_FREE(stIveFg.u32PhyAddr[0], stIveFg.pu8VirAddr[0]);
         IVE_MMZ_FREE(stIveBg.u32PhyAddr[0], stIveBg.pu8VirAddr[0]);
	 IVE_MMZ_FREE(pstDst->u32PhyAddr, pstDst->pu8VirAddr);
	if(HI_NULL != pstDst->pu8VirAddr)
	{
		free(pstDst->pu8VirAddr);
	}
	if(HI_NULL != pstDst)
	{
		free(pstDst);
	}

	if(HI_NULL != pfgRGB)
		cvReleaseImage(&pfgRGB);
	if(HI_NULL != pfg)
		cvReleaseImage(&pfg);
	if(HI_NULL != pBgGray)
		cvReleaseImage(&pBgGray);
	HI_MPI_SYS_Exit();
	
	return;
#endif
}

0xA01D8003  HI_ERR_IVE_ILLEGAL_PARAM 参数超出合法范围

看上去很好解决的样子，可是我就是不知道该怎么改