787
视频前景提取 (II)【IplImage版本】
以下代码可以求出视频的前景，用的方法就是拿第一帧做差分。。。
但是以下代码有个致命的问题，那就是在视频帧处理的过程中，不断的create却不释放内存，导致了内存溢出。。
#include "cv.h"
#include "highgui.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <math.h>
#include <float.h>
#include <limits.h>
#include <time.h>
#include <ctype.h>
#ifdef _EiC
#define WIN32
#endif
static CvMemStorage* storage = 0;
int main()
{
CvCapture* capture = 0;
IplImage *frame, *frame_copy = 0,*img1=NULL;
char* input_name;
input_name = "002.avi";
storage = cvCreateMemStorage(0);
if( !input_name || (isdigit(input_name[0]) && input_name[1] == '\0') )
//直接从摄像头读入
capture = cvCaptureFromCAM( !input_name ? 0 : input_name[0] - '0' );
else
//从AVI视频读入
capture = cvCaptureFromAVI( input_name );
cvNamedWindow( "result", 1 );
int flag=0;
if( capture )
{
//如果捕捉到帧了
//对每一帧做处理
for(;;)
{
if( !cvGrabFrame( capture ))
break;
frame = cvRetrieveFrame( capture );
if( !frame )
break;
if(flag==0)
{
//捕捉第一帧
frame_copy = cvCreateImage( cvSize(frame->width,frame->height), IPL_DEPTH_8U, frame->nChannels );
cvCopy(frame,frame_copy,0);
flag=1;
}
img1 = cvCreateImage( cvSize(frame->width,frame->height), IPL_DEPTH_8U, frame->nChannels );
cvAbsDiff(frame,frame_copy,img1);
cvShowImage( "src", frame );
cvShowImage( "result", img1 );
//10ms中按任意键进入此if块
if( cvWaitKey( 10 ) >= 0 )
break;
}
cvReleaseImage( &frame_copy );
cvReleaseImage( &img1 );
cvReleaseCapture( &capture );
}
cvDestroyWindow("src");
cvDestroyWindow("result");
return 0;
}
为了解决这个问题，可以在开始的时候设置标志位，只在第一次的时候create图片：
#include "cv.h"
#include "highgui.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <math.h>
#include <float.h>
#include <limits.h>
#include <time.h>
#include <ctype.h>
#ifdef _EiC
#define WIN32
#endif
int main()
{
CvCapture* capture = 0;
IplImage *frame, *frame_copy = 0,*img1=NULL;
char* input_name;
input_name = "001.avi";
if( !input_name || (isdigit(input_name[0]) && input_name[1] == '\0') )
//直接从摄像头读入
capture = cvCaptureFromCAM( !input_name ? 0 : input_name[0] - '0' );
else
//从AVI视频读入
capture = cvCaptureFromAVI( input_name );
cvNamedWindow( "result", 1 );
int flag=0;
if( capture )
{
//如果捕捉到帧了
//对每一帧做处理
for(;;)
{
if( !cvGrabFrame( capture ))
break;
frame = cvRetrieveFrame( capture );
if( !frame )
break;
if(flag==0)
{
//捕捉第一帧
if( !frame_copy )
frame_copy = cvCreateImage( cvSize(frame->width,frame->height), IPL_DEPTH_8U, frame->nChannels );
if( frame->origin == IPL_ORIGIN_TL )
cvCopy( frame, frame_copy, 0 );
else
cvFlip( frame, frame_copy, 0 );
flag=1;
}
//为NULL就create一次，不然内存会溢出
if( !img1 )
img1 = cvCreateImage( cvSize(frame->width,frame->height), IPL_DEPTH_8U, frame->nChannels );
cvAbsDiff(frame,frame_copy,img1);
cvShowImage( "src", frame );
cvShowImage( "result", img1 );
//10ms中按任意键进入此if块
if( cvWaitKey( 10 ) >= 0 )
break;
}
cvReleaseImage( &frame_copy );
cvReleaseCapture( &capture );
//不用释放，因为开始的时候没有显式分配内存
//cvReleaseImage( &frame );
//cvReleaseImage( &img1 );
}
cvDestroyWindow("src");
cvDestroyWindow("result");
return 0;
}
最后更新：2017-04-03 05:39:31