mirror of
https://github.com/opencv/opencv_contrib.git
synced 2025-10-17 07:04:18 +08:00
ced5aa7606
Implementation of CSR-DCF tracker (#1552) * Initial commit for CSR-DCF tracker implementation * Fixes for automatic build * General code fixes * Removed unused parameters. Added CSRT to automatic tests. * Fixed VS build warnings. Fixed a bug with gray sequences. * Fixed VS build errors for samples file.
33 lines
921 B
C++
33 lines
921 B
C++
#ifndef _SAMPLES_UTILITY_HPP_
|
|
#define _SAMPLES_UTILITY_HPP_
|
|
|
|
#include <opencv2/tracking.hpp>
|
|
|
|
inline cv::Ptr<cv::Tracker> createTrackerByName(cv::String name)
|
|
{
|
|
cv::Ptr<cv::Tracker> tracker;
|
|
|
|
if (name == "KCF")
|
|
tracker = cv::TrackerKCF::create();
|
|
else if (name == "TLD")
|
|
tracker = cv::TrackerTLD::create();
|
|
else if (name == "BOOSTING")
|
|
tracker = cv::TrackerBoosting::create();
|
|
else if (name == "MEDIAN_FLOW")
|
|
tracker = cv::TrackerMedianFlow::create();
|
|
else if (name == "MIL")
|
|
tracker = cv::TrackerMIL::create();
|
|
else if (name == "GOTURN")
|
|
tracker = cv::TrackerGOTURN::create();
|
|
else if (name == "MOSSE")
|
|
tracker = cv::TrackerMOSSE::create();
|
|
else if (name == "CSRT")
|
|
tracker = cv::TrackerCSRT::create();
|
|
else
|
|
CV_Error(cv::Error::StsBadArg, "Invalid tracking algorithm name\n");
|
|
|
|
return tracker;
|
|
}
|
|
|
|
#endif
|