#include <stdio.h> #include <windows.h> #include <cv.h> #include <highgui.h> int main (int argc, char **argv) { CvCapture *capture; int i, j, dx, dy, rows, cols; int block_size = 10; int shift_size = 1; CvMat *velx, *vely; CvSize block = cvSize (block_size, block_size); CvSize shift = cvSize (shift_size, shift_size); CvSize max_range = cvSize (50, 50); IplImage *src_img1, *src_img2, *dst_img; IplImage *tmp; capture = cvCreateCameraCapture(0); cvNamedWindow ("Image", 1); // (src_img1 = cvLoadImage (argv[1], CV_LOAD_IMAGE_GRAYSCALE)) == 0 || //(src_img2 = cvLoadImage (argv[2], CV_LOAD_IMAGE_GRAYSCALE)) == 0) dst_img = cvQueryFrame(capture);//カメラから画像を取り込む tmp = cvCreateImage(cvGetSize(dst_img), IPL_DEPTH_8U, 3); src_img1 = cvCreateImage(cvGetSize(dst_img), IPL_DEPTH_8U, 1); src_img2 = cvCreateImage(cvGetSize(dst_img), IPL_DEPTH_8U, 1); cvShowImage ("Image", dst_img); //dst_img = cvLoadImage (argv[2], CV_LOAD_IMAGE_COLOR); while(1){ // (1)速度ベクトルを格納する構造体の確保 rows = int (ceil (double (src_img1->height) / block_size)); cols = int (ceil (double (src_img1->width) / block_size)); velx = cvCreateMat (rows, cols, CV_32FC1); vely = cvCreateMat (rows, cols, CV_32FC1); cvSetZero (velx); cvSetZero (vely); dst_img = cvQueryFrame(capture);//カメラから画像を取り込む cvCvtColor(dst_img,src_img1,CV_BGR2GRAY); cvCopy(dst_img,tmp); // (2)オプティカルフローの計算 cvCalcOpticalFlowBM (src_img1, src_img2, block, shift, max_range, 0, velx, vely); // (3)計算されたフローを描画 for (i = 0; i < velx->width; i++) { for (j = 0; j < vely->height; j++) { dx = (int) cvGetReal2D (velx, j, i); dy = (int) cvGetReal2D (vely, j, i); cvLine (dst_img, cvPoint (i * block_size, j * block_size), cvPoint (i * block_size + dx, j * block_size + dy), CV_RGB (255, 0, 0), 1, CV_AA, 0); } } cvShowImage ("Image", dst_img); cvCvtColor(tmp,src_img2,CV_BGR2GRAY); //printf("%d\n",__LINE__); int key=cvWaitKey (10); if(key>=0)break; // cvReleaseImage (&dst_img); cvReleaseMat (&velx); cvReleaseMat (&vely); } cvDestroyWindow ("Image"); cvReleaseImage (&src_img1); cvReleaseImage (&src_img2); cvReleaseImage (&tmp); // cvReleaseImage (&dst_img); cvReleaseMat (&velx); cvReleaseMat (&vely); cvReleaseCapture(&capture); //終了後に固まるのはこれがないことが多い return 0; }