Hi,
Is there any suggestion to enhance the performance for this optical flow implementation
#define MAX_COUNT 250 #define DELAY_T 20 #definePI 3.1415 void main (){ IplImage* image = 0;
IplImage* current_Img = 0;
IplImage* Old_Img = 0;
IplImage * imgA = 0;
IplImage * imgB = 0;
CvCapture * capture = cvCreateFileCapture("1.mov");
cvNamedWindow("Origin");
IplImage * eig_image = 0;
IplImage * tmp_image = 0;
int corner_count = MAX_COUNT;
CvPoint2D32f* cornersA = new CvPoint2D32f[MAX_COUNT];
CvPoint2D32f * cornersB = new CvPoint2D32f[MAX_COUNT];
CvSize img_sz;
int win_size = 20;
IplImage* pyrA = 0;
IplImage* pyrB = 0;
char features_found[MAX_COUNT];
float feature_errors[MAX_COUNT];
int one_zero = 0;
int t_delay = 0;
while (1) {
if (cvGrabFrame(capture) == 0)
break;
if (Old_Img == 0)
{
image = cvRetrieveFrame(capture);
current_Img = cvCreateImage(cvSize(image->width, image->height), image->depth, image->nChannels);
Old_Img = cvCreateImage(cvSize(image->width, image->height), image->depth, image->nChannels);
}
if (one_zero == 0)
{
if (eig_image == 0)
{
eig_image = cvCreateImage(cvSize(image->width, image->height), image->depth, image->nChannels);
tmp_image = cvCreateImage(cvSize(image->width, image->height), image->depth, image->nChannels);
}
memcpy(Old_Img->imageData, current_Img->imageData, sizeof(char)*image->imageSize);
image = cvRetrieveFrame(capture);
memcpy(current_Img->imageData, image->imageData, sizeof(char)*image->imageSize);
if (imgA == 0)
{
imgA = cvCreateImage(cvSize(image->width, image->height), IPL_DEPTH_8U, 1);
imgB = cvCreateImage(cvSize(image->width, image->height), IPL_DEPTH_8U, 1);
}
cvCvtColor(current_Img, imgA, CV_BGR2GRAY);
cvCvtColor(Old_Img, imgB, CV_BGR2GRAY);
cvGoodFeaturesToTrack(imgA, eig_image, tmp_image, cornersA, &corner_count, 0.01, 5.0, 0, 3, 0, 0.04);
cvFindCornerSubPix(imgA, cornersA, corner_count, cvSize(win_size, win_size), cvSize(-1, -1), cvTermCriteria(CV_TERMCRIT_ITER | CV_TERMCRIT_EPS, 20, 0.03));
CvSize pyr_sz = cvSize(imgA->width + 8, imgB->height / 3);
if (pyrA == 0)
{
pyrA = cvCreateImage(pyr_sz, IPL_DEPTH_32F, 1);
pyrB = cvCreateImage(pyr_sz, IPL_DEPTH_32F, 1);
}
cvCalcOpticalFlowPyrLK(imgA, imgB, pyrA, pyrB, cornersA, cornersB, corner_count, cvSize(win_size, win_size), 5, features_found, feature_errors, cvTermCriteria(CV_TERMCRIT_ITER | CV_TERMCRIT_EPS, 20, 0.3), 0);
for (int i = 0; i< corner_count; ++i)
{
if (features_found[i] == 0 || feature_errors[i] > MAX_COUNT)
continue;
float fVecLength = sqrt((float)((cornersA[i].x - cornersB[i].x)*(cornersA[i].x - cornersB[i].x) + (cornersA[i].y - cornersB[i].y)*(cornersA[i].y - cornersB[i].y)));
//Vector Angle
float fVecSetha = fabs(atan2((float)(cornersB[i].y - cornersA[i].y), (float)(cornersB[i].x - cornersA[i].x)) * 180 / PI);
cvLine(image, cvPoint(cornersA[i].x, cornersA[i].y), cvPoint(cornersB[i].x, cornersA[i].y), CV_RGB(0, 255, 0), 2);
printf("[%d] - Sheta:%lf, Length:%lf\n", i, fVecSetha, fVecLength);
}
}
cvShowImage("Origin", image);
one_zero++;
if ((one_zero % DELAY_T) == 0)
{
one_zero = 0;
}
if (cvWaitKey(10) >= 0)
break;
}
cvReleaseCapture(&capture);
cvDestroyWindow("Origin");
cvReleaseImage(&Old_Img);
cvReleaseImage(&imgA);
cvReleaseImage(&imgB);
cvReleaseImage(&eig_image);
cvReleaseImage(&tmp_image);
delete cornersA;
delete cornersB;
cvReleaseImage(&pyrA);
cvReleaseImage(&pyrB);
}
↧