void MedianFlowTracker::track(const Mat& prevMat, const Mat& currMat, Rect* prevBB) {
if (prevBB != NULL) {
if (prevBB->width <= 0 || prevBB->height <= 0) {
return;
}
float bb_tracker[] = {
float(prevBB->x),
float(prevBB->y),
float(prevBB->width + prevBB->x - 1),
float(prevBB->height + prevBB->y - 1)
};
float scale;
IplImage prevImg = prevMat;
IplImage currImg = currMat;
int success = fbtrack(&prevImg, &currImg, bb_tracker, bb_tracker, &scale);
//Extract subimage
float x, y, w, h;
x = floor(bb_tracker[0] + 0.5);
y = floor(bb_tracker[1] + 0.5);
w = floor(bb_tracker[2] - bb_tracker[0] + 1 + 0.5);
h = floor(bb_tracker[3] - bb_tracker[1] + 1 + 0.5);
//TODO: Introduce a check for a minimum size
if (!success || x < 0 || y < 0 || w <= 0 || h <= 0 || x + w > currMat.cols || y + h > currMat.rows || x != x || y != y || w != w || h != h) { //x!=x is check for nan
//Leave it empty
} else {
trackerBB = new Rect(x, y, w, h);
}
}
}
void MedianFlowTracker::track(Mat prevMat, Mat currMat, Rect* prevBB) {
if(prevBB != NULL) {
if(prevBB->width <= 0 || prevBB->height <= 0) {
return;
}
float bb_tracker[] = {prevBB->x, prevBB->y, prevBB->width+prevBB->x-1, prevBB->height+prevBB->y-1};
float scale;
IplImage prevImg = prevMat;
IplImage currImg = currMat;
float cosine = 0, magnitude = 0;
int success = fbtrack(&prevImg, &currImg, bb_tracker,bb_tracker,&scale, &cosine, &magnitude);
//Extract subimage
float x,y,w,h;
x = floor(bb_tracker[0]+0.5);
y = floor(bb_tracker[1]+0.5);
w = floor(bb_tracker[2]-bb_tracker[0]+1+0.5);
h = floor(bb_tracker[3]-bb_tracker[1]+1+0.5);
//TODO: Introduce a check for a minimum size
if(!success || x < 0 || y < 0 || w <= 0 || h <= 0 || x +w > currMat.cols || y+h > currMat.rows || x!=x || y!=y || w!=w || h!=h) { //x!=x is check for nan
//Leave it empty
} else {
trackerBB = new Rect(x,y,w,h);
//printf("BBinside %d %d %d %d\n", trackerBB->x, trackerBB->y, trackerBB->width, trackerBB->height);
}
this->cosine = cosine;
this->magnitude = magnitude;
}
}
