mirror of
https://github.com/opencv/opencv_contrib.git
synced 2025-10-18 17:24:28 +08:00
31 lines
847 B
C++
31 lines
847 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
|
|
CV_Error(cv::Error::StsBadArg, "Invalid tracking algorithm name\n");
|
|
|
|
return tracker;
|
|
}
|
|
|
|
#endif
|