金字塔Lucas-Kanande光流算法实现

// Lucas-Kanade method Optical Flow in OpenCV
// BJTShang, 2016-12-13

#include <cv.h>
#include <cxcore.h>
#include <highgui.h>

const int MAX_CORNERS = 500;

int main(int argc, char** argv){
  IplImage* imgA = cvLoadImage("/home/bjtshang/cpp_ws/opencv/data/OpticalFlow0.jpg",
			       CV_LOAD_IMAGE_GRAYSCALE);
  IplImage* imgB = cvLoadImage("/home/bjtshang/cpp_ws/opencv/data/OpticalFlow1.jpg",
			       CV_LOAD_IMAGE_GRAYSCALE);
  
  // image to show the optical flow vectors
  IplImage* imgC = cvLoadImage("/home/bjtshang/Desktop/OpticalFlow1.jpg",
			       CV_LOAD_IMAGE_UNCHANGED);
  
  CvSize img_size = cvGetSize(imgA);
  CvSize win_size = cvSize(50, 50);
  int corner_count = MAX_CORNERS;
  
  // get the features (detect corners) need to be tracked
  IplImage* imgEig = cvCreateImage(img_size, IPL_DEPTH_32F, 1);
  IplImage* imgTmp = cvCreateImage(img_size, IPL_DEPTH_32F, 1);
  
  CvPoint2D32f* cornersA = new CvPoint2D32f[corner_count];
  
  cvGoodFeaturesToTrack(imgA, imgEig, imgTmp, cornersA, &corner_count,
			0.02, 8.0, 0, 3, 0, 0.04);
  
  // find sub-pixel corners
  cvFindCornerSubPix(imgA, cornersA, corner_count, 
		     win_size, cvSize(-1,-1),
		     cvTermCriteria(CV_TERMCRIT_ITER|CV_TERMCRIT_EPS, 20, 0.1));
  
  char features_found[MAX_CORNERS];
  float feature_errors[MAX_CORNERS];
  
  CvSize pyr_size = cvSize(imgA->width+8, imgB->height/3);
  
  IplImage* pyrA = cvCreateImage(pyr_size, IPL_DEPTH_32F, 1);
  IplImage* pyrB = cvCreateImage(pyr_size, IPL_DEPTH_32F, 1);
  
  CvPoint2D32f* cornersB = new CvPoint2D32f[MAX_CORNERS];
  
  cvCalcOpticalFlowPyrLK(imgA, imgB, pyrA, pyrB, cornersA, cornersB,
			 corner_count, win_size,
			 10, features_found, feature_errors,
			 cvTermCriteria(CV_TERMCRIT_ITER|CV_TERMCRIT_EPS, 20, 0.1),
			 0
  );
  
  for(int i=0; i<corner_count; i++){
    if(features_found[i]==0 || feature_errors[i] > 550){
      printf("Error is: %f\n", feature_errors[i]);
      continue;
    }
    //printf("Got it\n");
    CvPoint p1 = cvPoint(cvRound(cornersA[i].x), cvRound(cornersA[i].y));
    CvPoint p2 = cvPoint(cvRound(cornersB[i].x), cvRound(cornersB[i].y));
    cvLine(imgC, p1, p2, CV_RGB(255, 0, 0), 1);
  }
  cvNamedWindow("imgA", 0);
  cvNamedWindow("imgB", 0);
  cvNamedWindow("Optical_flow", 0);
  cvShowImage("imgA", imgA);
  cvShowImage("imgB", imgB);
  cvShowImage("Optical_flow", imgC);
  
  cvWaitKey(0);
  return 0;
}