parent
b46d23a2aa
commit
7860d52e14
1 changed files with 63 additions and 0 deletions
@ -0,0 +1,63 @@ |
||||
#include <opencv2/video/background_segm.hpp> |
||||
#include <opencv2/highgui/highgui.hpp> |
||||
|
||||
#include <stdio.h> |
||||
|
||||
//this is a sample for foreground detection functions
|
||||
int main(int argc, char** argv) |
||||
{ |
||||
IplImage* tmp_frame = NULL; |
||||
CvCapture* cap = NULL; |
||||
bool update_bg_model = true; |
||||
|
||||
if( argc < 2 ) |
||||
cap = cvCaptureFromCAM(0); |
||||
else |
||||
cap = cvCaptureFromFile(argv[1]); |
||||
|
||||
if( !cap ) |
||||
{ |
||||
printf("can not open camera or video file\n"); |
||||
return -1; |
||||
} |
||||
|
||||
tmp_frame = cvQueryFrame(cap); |
||||
if(!tmp_frame) |
||||
{ |
||||
printf("can not read data from the video source\n"); |
||||
return -1; |
||||
} |
||||
|
||||
cvNamedWindow("BG", 1); |
||||
cvNamedWindow("FG", 1); |
||||
|
||||
CvBGStatModel* bg_model = 0; |
||||
|
||||
for( int fr = 1;tmp_frame; tmp_frame = cvQueryFrame(cap), fr++ ) |
||||
{ |
||||
if(!bg_model) |
||||
{ |
||||
//create BG model
|
||||
bg_model = cvCreateGaussianBGModel( tmp_frame ); |
||||
//bg_model = cvCreateFGDStatModel( temp );
|
||||
continue; |
||||
} |
||||
|
||||
double t = (double)cvGetTickCount(); |
||||
cvUpdateBGStatModel( tmp_frame, bg_model, update_bg_model ? -1 : 0 ); |
||||
t = (double)cvGetTickCount() - t; |
||||
printf( "%d. %.1f\n", fr, t/(cvGetTickFrequency()*1000.) ); |
||||
cvShowImage("BG", bg_model->background); |
||||
cvShowImage("FG", bg_model->foreground); |
||||
char k = cvWaitKey(5); |
||||
if( k == 27 ) break; |
||||
if( k == ' ' ) |
||||
update_bg_model = !update_bg_model; |
||||
} |
||||
|
||||
|
||||
cvReleaseBGStatModel( &bg_model ); |
||||
cvReleaseCapture(&cap); |
||||
|
||||
return 0; |
||||
} |
Loading…
Reference in new issue