opencv码本(codebook)背景模型

/结构体定义 typedef struct CvBGCodeBookElem struct CvBGCodeBookElem* next; int tLastUpdate; int stale; uchar boxMin3; uchar boxMax3; uchar learnMin3; uchar learnMax3; CvBGCodeBookElem; typedef struct CvBGCodeBookModel CvSize size; int t; uchar cbBounds3; uchar modMin3; uchar modMax3; CvBGCodeBookElem* cbmap; CvMemStorage* storage; CvBGCodeBookElem* freeList; CvBGCodeBookModel; CvBGCodeBookModel* cvCreateBGCodeBookModel() CvBGCodeBookModel* model = (CvBGCodeBookModel*)cvAlloc( sizeof(*model) ); memset( model, 0, sizeof(*model) ); model-cbBounds0 = model-cbBounds1 = model-cbBounds2 = 10; model-modMin0 = 3; model-modMax0 = 10; model-modMin1 = model-modMin2 = 1; model-modMax1 = model-modMax2 = 1; model-storage = cvCreateMemStorage(); return model; void cvReleaseBGCodeBookModel( CvBGCodeBookModel* model ) if( model memset( *model, 0, sizeof(*model) ); cvFree( model ); static uchar satTab8u768; #undef SAT_8U #define SAT_8U(x) satTab8u(x) + 255 static void icvInitSatTab() static int initialized = 0; if( !initialized ) for( int i = 0; i 255 ? 255 : v); initialized = 1; void cvBGCodeBookUpdate( CvBGCodeBookModel* model,const CvArr* _image, CvRect roi=cvRect(0,0,0,0), const CvArr* _mask=0 ) CV_FUNCNAME( “cvBGCodeBookUpdate“ ); _BEGIN_; CvMat stub, *image = cvGetMat( _image, CvMat mstub, *mask = _mask ? cvGetMat( _mask, int i, x, y, T; int nblocks; uchar cb0, cb1, cb2; CvBGCodeBookElem* freeList; CV_ASSERT( model if( roi.x = 0 roi.height = image-rows; else CV_ASSERT( (unsigned)roi.x cols if( image-cols != model-size.width | image-rows != model-size.height ) cvClearMemStorage( model-storage ); model-freeList = 0; cvFree( int bufSz = image-cols*image-rows*sizeof(model-cbmap0); model-cbmap = (CvBGCodeBookElem*)cvAlloc(bufSz); memset( model-cbmap, 0, bufSz ); model-size = cvSize(image-cols, image-rows); icvInitSatTab(); cb0 = model-cbBounds0; cb1 = model-cbBounds1; cb2 = model-cbBounds2; T = +model-t; freeList = model-freeList; nblocks = (int)(model-storage-block_size - sizeof(CvMemBlock)/sizeof(*freeList); nblocks = MIN( nblocks, 1024 ); CV_ASSERT( nblocks 0 ); for( y = 0; y data.ptr + image-step*(y + roi.y) + roi.x*3; const uchar* m = mask ? mask-data.ptr + mask-step*(y + roi.y) + roi.x : 0; CvBGCodeBookElem* cb = model-cbmap + image-cols*(y + roi.y) + roi.x; for( x = 0; x next ) if( e-learnMin0 learnMax0 color=#FF0000/此处修改 */color (e-stale)+; e-boxMin0 = MIN(e-boxMin0, p0); e-boxMax0 = MAX(e-boxMax0, p0); e-boxMin1 = MIN(e-boxMin1, p1); e-boxMax1 = MAX(e-boxMax1, p1); e-boxMin2 = MIN(e-boxMin2, p2); e-boxMax2 = MAX(e-boxMax2, p2); / no need to use SAT_8U for updated learnMini if( e-learnMax0 learnMax0+; if( e-learnMin1 l1 ) e-learnMin1-; if( e-learnMax1 learnMax1+; if( e-learnMin2 l2 ) e-learnMin2-; if( e-learnMax2 learnMax2+; found = e; break; /negRun = T - e-tLastUpdate;color=#FF0000/此处修改 *8/color / e-stale = MAX( e-stale, negRun ); /*for( ; e != 0; e = e-next )color=#FF0000/此处修改 * */color negRun = T - e-tLastUpdate; e-stale = MAX( e-stale, negRun ); */ if( !found ) if( !freeList ) freeList = (CvBGCodeBookElem*)cvMemStorageAlloc(model-storage, nblocks*sizeof(*freeList); for( i = 0; i next; e-learnMin0 = l0; e-learnMax0 = h0; e-learnMin1 = l1; e-learnMax1 = h1; e-learnMin2 = l2; e-learnMax2 = h2; e-boxMin0 = e-boxMax0 = p0; e-boxMin1 = e-boxMax1 = p1; e-boxMin2 = e-boxMax2 = p2; /e-tLastUpdate = T; e-stale=1;color=#FF0000/此处修改 * /color e-stale = 0; e-next = *cb; *cb = e; model-freeList = freeList; _END_; int cvBGCodeBookDiff( const CvBGCodeBookModel* model, const CvArr* _image, CvArr* _fgmask, CvRect roi=cvRect(0,0,0,0) int maskCount = -1; CV_FUNCNAME( “cvBGCodeBookDiff“ ); _BEGIN_; CvMat stub, *image = cvGetMat( _image, CvMat mstub, *mask = cvGetMat( _fgmask, int x, y; uchar m0,m1,m2,M0,M1,M2; CV_ASSERT( model if( roi.x = 0 roi.height = image-rows; else CV_ASSERT( (unsigned)roi.x cols m0 = model-modMin0; M0 = model-modMax0; m1 = model-modMin1; M1 = model-modMax1; m2 = model-modMin2; M2 = model-modMax2; maskCount = roi.height*roi.width; for( y = 0; y data.ptr + image-step*(y + roi.y) + roi.x*3; uchar* m = mask-data.ptr + mask-step*(y + roi.y) + roi.x; CvBGCodeBookElem* cb = model-cbmap + image-cols*(y + roi.y) + roi.x; for( x = 0; x next ) if( e-boxMin0 boxMax0 maskCount-; break; _END_; return maskCount; void cvBGCodeBookClearStale( CvBGCodeBookModel* model, int staleThresh, CvRect roi=cvRect(0,0,0,0), const CvArr* _mask=0 ) CV_FUNCNAME( “cvBGCodeBookClearStale“ ); _BEGIN_; CvMat mstub, *mask = _mask ? cvGetMat( _mask, int x, y, T; CvBGCodeBookElem* freeList; CV_ASSERT( model if( roi.x = 0 roi.height = model-size.height; else CV_ASSERT( (unsigned)roi.x cols icvInitSatTab(); freeList = model-freeList; T = model-t; for( y = 0; y data.ptr + mask-step*(y + roi.y) + roi.x : 0; CvBGCodeBookElem* cb =