-
Notifications
You must be signed in to change notification settings - Fork 18
/
CVFFarneback.m
89 lines (76 loc) · 2.45 KB
/
CVFFarneback.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
//
// CVFFarneback.m
// CVFunhouse
//
// Created by John Brewer on 7/22/12.
// Copyright (c) 2012 Jera Design LLC. All rights reserved.
//
// Based on the OpenCV example: <opencv>/samples/c/fback_c.c
#import "CVFFarneback.h"
#include "opencv2/video/tracking.hpp"
#include "opencv2/imgproc/imgproc_c.h"
static void drawOptFlowMap(const CvMat* flow, CvMat* cflowmap, int step,
double scale, CvScalar color)
{
int x, y;
(void)scale;
for( y = 0; y < cflowmap->rows; y += step)
for( x = 0; x < cflowmap->cols; x += step)
{
CvPoint2D32f fxy = CV_MAT_ELEM(*flow, CvPoint2D32f, y, x);
cvLine(cflowmap, cvPoint(x,y), cvPoint(cvRound(x+fxy.x), cvRound(y+fxy.y)),
color, 1, 8, 0);
cvCircle(cflowmap, cvPoint(x,y), 2, color, -1, 8, 0);
}
}
@interface CVFFarneback () {
CvMat *prevgray;
CvMat *gray;
CvMat *flow;
CvMat *cflow;
}
@end
@implementation CVFFarneback
/*
* processIplImage
*
* Inputs:
* iplImage: an IplImage in BGRA format, 8 bits per pixel.
* YOU ARE RESPONSIBLE FOR CALLING cvReleaseImage on this image.
*
* Outputs:
* When you are done, call imageReady: with an RGB, RGBA, or grayscale
* IplImage with 8-bits per pixel.
*
* You can call imageReady: from any thread and it will do the right thing.
* You can fork as many threads to process the image as you like; just call
* imageReady when you are done.
*
* imageReady: will dispose of the IplImage you pass it once the system is
* done with it.
*/
-(void)processIplImage:(IplImage*)frame
{
int firstFrame = (gray == 0);
if(!gray)
{
gray = cvCreateMat(frame->height, frame->width, CV_8UC1);
prevgray = cvCreateMat(gray->rows, gray->cols, gray->type);
flow = cvCreateMat(gray->rows, gray->cols, CV_32FC2);
cflow = cvCreateMat(gray->rows, gray->cols, CV_8UC3);
}
cvCvtColor(frame, gray, CV_BGR2GRAY);
cvReleaseImage(&frame);
cvCvtColor(gray, cflow, CV_GRAY2BGR);
if( !firstFrame )
{
cvCalcOpticalFlowFarneback(prevgray, gray, flow, 0.5, 3, 15, 3, 5, 1.2, 0);
drawOptFlowMap(flow, cflow, 16, 1.5, CV_RGB(0, 255, 0));
}
cvCopy(gray, prevgray, nil);
// Call imageReady with your new image.
IplImage *tempImage = cvAlloc(sizeof(IplImage));
IplImage *outImage = cvGetImage(cflow, tempImage);
[self imageReady: outImage];
}
@end