#include <opencv2/core.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/tracking/tracker.hpp>
#include <opencv2/core/utility.hpp>
#include <opencv2/tracking.hpp>
#include <opencv2/videoio.hpp>
#include <iostream>
#include <vector>
using namespace cv;
void mouseClickCallback(int event,
int x, int y, int flags, void* userdata)
{
cv::Rect2d * pRect =
reinterpret_cast<cv::Rect2d*>(userdata);
if (event == cv::EVENT_LBUTTONDOWN)
{
std::cout << "LBUTTONDOWN ("
<< x << ", " << y << ")" << std::endl;
pRect->x = x;
pRect->y = y;
}
else if (event == cv::EVENT_LBUTTONUP)
{
std::cout << "LBUTTONUP ("
<< x << ", " << y << ")" << std::endl;
pRect->width = std::abs(x - pRect->x);
pRect->height = std::abs(y - pRect->y);
}
}
int main(int argc, char** argv)
{
cv::VideoCapture cap("..\\images\\car.avi");
if (!cap.isOpened())
{
std::cout << " on data! " << std::endl;
return -1;
}
cap.set(CV_CAP_PROP_POS_MSEC, 2*1000);
cv::Mat frame;
cv::Rect2d *rect(new cv::Rect2d);
cv::Ptr<cv::TrackerTLD> tracker =
cv::TrackerTLD::createTracker();
cap >> frame;
cv::resize(frame, frame, cv::Size(), 0.25,0.25);
cv::imshow("TrackerTLD", frame);
cv::setMouseCallback("TrackerTLD", mouseClickCallback,
reinterpret_cast<void*>(rect));
cv::waitKey();
if (rect->area() == 0.0)
return -1;
tracker->init(frame, *rect);
double fps = 1.0;
while(true)
{
cap >> frame;
cv::resize(frame, frame, cv::Size(), 0.25,0.25);
if (frame.empty())
break;
if (tracker->update(frame, *rect))
cv::rectangle(frame, *rect,
cv::Scalar(255, 0, 0 ), 2, 1);
cv::imshow("TrackerTLD", frame);
char c = cv::waitKey(10);
if (27 == c)
break;
}
cap.release();
return 0;
}