Mercurial > hg > nsaunier > traffic-intelligence

comparison c/optical-flow.cpp @ 15:3ead4bcd001c

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
cleaned optical flow
author Nicolas Saunier <nico@confins.net>
date Sun, 15 Nov 2009 01:04:10 -0500
parents ff5403319cec
children bc4ea09b1743
comparison
equal deleted inserted replaced
14:e7bbe8465591 15:3ead4bcd001c
68 IplImage* pyramid1 = ::allocateImage(frameSize, IPL_DEPTH_8U, 1); 68 IplImage* pyramid1 = ::allocateImage(frameSize, IPL_DEPTH_8U, 1);
69 IplImage* pyramid2 = ::allocateImage(frameSize, IPL_DEPTH_8U, 1); 69 IplImage* pyramid2 = ::allocateImage(frameSize, IPL_DEPTH_8U, 1);
70 70
71 int pressedKey = '?'; 71 int pressedKey = '?';
72 while (frame && !::interruptionKey(pressedKey)) { 72 while (frame && !::interruptionKey(pressedKey)) {
73 cvGoodFeaturesToTrack(frame1_1C, eig_image, temp_image, frame1_features, &nFeatures, .01, .01, NULL); 73 cvGoodFeaturesToTrack(frame1_1C, eig_image, temp_image, frame1_features, &nFeatures, 0.05 /*quality*/, 1/* min dist*/, NULL);
74 cvCalcOpticalFlowPyrLK(frame1_1C, frame2_1C, pyramid1, pyramid2, frame1_features, frame2_features, nFeatures, optical_flow_window, 5, optical_flow_found_feature, optical_flow_feature_error, optical_flow_termination_criteria, 0 ); 74 cvCalcOpticalFlowPyrLK(frame1_1C, frame2_1C, pyramid1, pyramid2, frame1_features, frame2_features, nFeatures, optical_flow_window, 5, optical_flow_found_feature, optical_flow_feature_error, optical_flow_termination_criteria, 0 );
75 75
76 for(int i = 0; i < nFeatures; i++) { 76 for(int i = 0; i < nFeatures; i++) {
77 /* If Pyramidal Lucas Kanade didn't really find the feature, skip it. */ 77 if (optical_flow_found_feature[i] != 0) {
78 if ( optical_flow_found_feature[i] == 0 ) continue; 78 int line_thickness = 1;
79 CvScalar line_color = CV_RGB(255,0,0);
79 80
80 int line_thickness; line_thickness = 1; 81 CvPoint p,q;
81 /* CV_RGB(red, green, blue) is the red, green, and blue components 82 p.x = lrintf(frame1_features[i].x);
82 * of the color you want, each out of 255. 83 p.y = lrintf(frame1_features[i].y);
83 */ 84 q.x = lrintf(frame2_features[i].x);
84 CvScalar line_color; line_color = CV_RGB(255,0,0); 85 q.y = lrintf(frame2_features[i].y);
86
87 double dx = frame2_features[i].x-frame1_features[i].x;
88 double dy = frame2_features[i].y-frame1_features[i].y;
89
90 double angle = atan2( (double) p.y - q.y, (double) p.x - q.x );
91 double dist = sqrt(square(dx) + square(dy));
92
93 if (dist > 2.) { // min motion distance
94 q.x = lrintf(frame1_features[i].x+dx*3); //(p.x - 3 * hypotenuse * cos(angle));
95 q.y = lrintf(frame1_features[i].y+dy*3); //(p.y - 3 * hypotenuse * sin(angle));
85 96
86 /* Let's make the flow field look nice with arrows. */ 97 /* Now we draw the main line of the arrow. */
87 98 cvLine( frame1, p, q, line_color, line_thickness, CV_AA, 0 );
88 /* The arrows will be a bit too short for a nice visualization because of the high framerate 99 /* Now draw the tips of the arrow. I do some scaling so that the
89 * (ie: there's not much motion between the frames). So let's lengthen them by a factor of 3. 100 * tips look proportional to the main line of the arrow.
90 */ 101 */
91 CvPoint p,q; 102 float arrowSize = 6;
92 p.x = (int) frame1_features[i].x; 103 p.x = lrintf(q.x + arrowSize * cos(angle + pi / 4));
93 p.y = (int) frame1_features[i].y; 104 p.y = lrintf(q.y + arrowSize * sin(angle + pi / 4));
94 q.x = (int) frame2_features[i].x; 105 cvLine( frame1, p, q, line_color, line_thickness, CV_AA, 0 );
95 q.y = (int) frame2_features[i].y; 106 p.x = lrintf(q.x + arrowSize * cos(angle - pi / 4));
96 107 p.y = lrintf(q.y + arrowSize * sin(angle - pi / 4));
97 double angle; angle = atan2( (double) p.y - q.y, (double) p.x - q.x ); 108 cvLine( frame1, p, q, line_color, line_thickness, CV_AA, 0 );
98 double hypotenuse; hypotenuse = sqrt( square(p.y - q.y) + square(p.x - q.x) ); 109 }
99 110 }
100 /* Here we lengthen the arrow by a factor of three. */
101 q.x = (int) (p.x - 3 * hypotenuse * cos(angle));
102 q.y = (int) (p.y - 3 * hypotenuse * sin(angle));
103
104 /* Now we draw the main line of the arrow. */
105 /* "frame1" is the frame to draw on.
106 * "p" is the point where the line begins.
107 * "q" is the point where the line stops.
108 * "CV_AA" means antialiased drawing.
109 * "0" means no fractional bits in the center cooridinate or radius.
110 */
111 cvLine( frame1, p, q, line_color, line_thickness, CV_AA, 0 );
112 /* Now draw the tips of the arrow. I do some scaling so that the
113 * tips look proportional to the main line of the arrow.
114 */
115 p.x = (int) (q.x + 9 * cos(angle + pi / 4));
116 p.y = (int) (q.y + 9 * sin(angle + pi / 4));
117 cvLine( frame1, p, q, line_color, line_thickness, CV_AA, 0 );
118 p.x = (int) (q.x + 9 * cos(angle - pi / 4));
119 p.y = (int) (q.y + 9 * sin(angle - pi / 4));
120 cvLine( frame1, p, q, line_color, line_thickness, CV_AA, 0 );
121 } 111 }
122 cvShowImage("Optical Flow", frame1); 112 cvShowImage("Optical Flow", frame1);
123 //cvWaitKey(5);
124 pressedKey = cvWaitKey(5); 113 pressedKey = cvWaitKey(5);
125 frame = cvQueryFrame(inputVideo); 114 frame = cvQueryFrame(inputVideo);
126 frameNum++; 115 frameNum++;
127 cout << frameNum << endl; 116 cout << frameNum << endl;
128 117