opencv/opencv_contrib
1
0
Fork 0
mirror of https://github.com/opencv/opencv_contrib.git synced 2025-10-20 12:55:15 +08:00
Code Issues Releases Wiki Activity

tracking: rework tracking API

Browse Source 
- simplify Tracker API
- fix perf tests (don't measure video decoding)
This commit is contained in:
Alexander Alekhin
2020-10-27 21:18:18 +00:00
parent 2cb3f65131
commit 80c197590c
89 changed files with 3629 additions and 4334 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
modules/tracking/samples/samples_utility.hpp
View File

@@ -2,25 +2,28 @@
#define _SAMPLES_UTILITY_HPP_
#include <opencv2/tracking.hpp>
#include <opencv2/tracking/tracking_legacy.hpp>
inline cv::Ptr<cv::Tracker> createTrackerByName(cv::String name)
inline cv::Ptr<cv::Tracker> createTrackerByName(const std::string& name)
{
using namespace cv;
cv::Ptr<cv::Tracker> tracker;
if (name == "KCF")
tracker = cv::TrackerKCF::create();
else if (name == "TLD")
tracker = cv::TrackerTLD::create();
tracker = legacy::upgradeTrackingAPI(legacy::TrackerTLD::create());
else if (name == "BOOSTING")
tracker = cv::TrackerBoosting::create();
tracker = legacy::upgradeTrackingAPI(legacy::TrackerBoosting::create());
else if (name == "MEDIAN_FLOW")
tracker = cv::TrackerMedianFlow::create();
tracker = legacy::upgradeTrackingAPI(legacy::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();
tracker = legacy::upgradeTrackingAPI(legacy::TrackerMOSSE::create());
else if (name == "CSRT")
tracker = cv::TrackerCSRT::create();
else
@@ -29,4 +32,32 @@ inline cv::Ptr<cv::Tracker> createTrackerByName(cv::String name)
return tracker;
}
inline cv::Ptr<cv::legacy::Tracker> createTrackerByName_legacy(const std::string& name)
{
using namespace cv;
cv::Ptr<cv::legacy::Tracker> tracker;
if (name == "KCF")
tracker = legacy::TrackerKCF::create();
else if (name == "TLD")
tracker = legacy::TrackerTLD::create();
else if (name == "BOOSTING")
tracker = legacy::TrackerBoosting::create();
else if (name == "MEDIAN_FLOW")
tracker = legacy::TrackerMedianFlow::create();
else if (name == "MIL")
tracker = legacy::TrackerMIL::create();
else if (name == "GOTURN")
CV_Error(cv::Error::StsNotImplemented, "FIXIT: migration on new API is required");
else if (name == "MOSSE")
tracker = legacy::TrackerMOSSE::create();
else if (name == "CSRT")
tracker = legacy::TrackerCSRT::create();
else
CV_Error(cv::Error::StsBadArg, "Invalid tracking algorithm name\n");
return tracker;
}
#endif