Vladimir
10 years ago
parent
1565ff1037
commit
fdc87172b8
3 changed files with 218 additions and 61 deletions
@ -0,0 +1,120 @@ |
||||
/*M///////////////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
||||
//
|
||||
// By downloading, copying, installing or using the software you agree to this license.
|
||||
// If you do not agree to this license, do not download, install,
|
||||
// copy or use the software.
|
||||
//
|
||||
//
|
||||
// License Agreement
|
||||
// For Open Source Computer Vision Library
|
||||
//
|
||||
// Copyright (C) 2013, OpenCV Foundation, all rights reserved.
|
||||
// Third party copyrights are property of their respective owners.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without modification,
|
||||
// are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistribution's of source code must retain the above copyright notice,
|
||||
// this list of conditions and the following disclaimer.
|
||||
//
|
||||
// * Redistribution's in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
//
|
||||
// * The name of the copyright holders may not be used to endorse or promote products
|
||||
// derived from this software without specific prior written permission.
|
||||
//
|
||||
// This software is provided by the copyright holders and contributors "as is" and
|
||||
// any express or implied warranties, including, but not limited to, the implied
|
||||
// warranties of merchantability and fitness for a particular purpose are disclaimed.
|
||||
// In no event shall the Intel Corporation or contributors be liable for any direct,
|
||||
// indirect, incidental, special, exemplary, or consequential damages
|
||||
// (including, but not limited to, procurement of substitute goods or services;
|
||||
// loss of use, data, or profits; or business interruption) however caused
|
||||
// and on any theory of liability, whether in contract, strict liability,
|
||||
// or tort (including negligence or otherwise) arising in any way out of
|
||||
// the use of this software, even if advised of the possibility of such damage.
|
||||
//
|
||||
//M*/
|
||||
|
||||
#include "precomp.hpp" |
||||
|
||||
namespace cv { |
||||
|
||||
// constructor
|
||||
MultiTracker::MultiTracker(const String& trackerType):defaultAlgorithm(trackerType){}; |
||||
|
||||
// destructor
|
||||
MultiTracker::~MultiTracker(){}; |
||||
|
||||
// add an object to be tracked, defaultAlgorithm is used
|
||||
bool MultiTracker::add(const Mat& image, const Rect2d& boundingBox){ |
||||
// quit if defaultAlgorithm has not been configured
|
||||
if(defaultAlgorithm==""){ |
||||
printf("Default algorithm was not defined!\n"); |
||||
return false; |
||||
} |
||||
|
||||
// add a new tracked object
|
||||
return add(defaultAlgorithm.c_str(), image, boundingBox); |
||||
}; |
||||
|
||||
// add a new tracked object
|
||||
bool MultiTracker::add( const String& trackerType, const Mat& image, const Rect2d& boundingBox ){ |
||||
// declare a new tracker
|
||||
Ptr<Tracker> newTracker = Tracker::create( trackerType ); |
||||
|
||||
// add the created tracker algorithm to the trackers list
|
||||
trackerList.push_back(newTracker); |
||||
|
||||
// add the ROI to the bounding box list
|
||||
objects.push_back(boundingBox); |
||||
|
||||
// initialize the created tracker
|
||||
return trackerList.back()->init(image, boundingBox); |
||||
}; |
||||
|
||||
// add a set of objects to be tracked
|
||||
bool MultiTracker::add(const String& trackerType, const Mat& image, std::vector<Rect2d> boundingBox){ |
||||
// status of the tracker addition
|
||||
bool stat=false; |
||||
|
||||
// add tracker for all input objects
|
||||
for(unsigned i =0;i<boundingBox.size();i++){ |
||||
stat=add(trackerType,image,boundingBox[i]); |
||||
if(!stat)break; |
||||
} |
||||
|
||||
// return the status
|
||||
return stat; |
||||
}; |
||||
|
||||
// add a set of object to be tracked, defaultAlgorithm is used.
|
||||
bool MultiTracker::add(const Mat& image, std::vector<Rect2d> boundingBox){ |
||||
// quit if defaultAlgorithm has not been configured
|
||||
if(defaultAlgorithm==""){ |
||||
printf("Default algorithm was not defined!\n"); |
||||
return false; |
||||
} |
||||
|
||||
return add(defaultAlgorithm.c_str(), image, boundingBox); |
||||
}; |
||||
|
||||
// update position of the tracked objects, the result is stored in internal storage
|
||||
bool MultiTracker::update( const Mat& image){ |
||||
for(unsigned i=0;i< trackerList.size(); i++){ |
||||
trackerList[i]->update(image, objects[i]); |
||||
} |
||||
return true; |
||||
}; |
||||
|
||||
// update position of the tracked objects, the result is copied to external variable
|
||||
bool MultiTracker::update( const Mat& image, std::vector<Rect2d> & boundingBox ){ |
||||
update(image); |
||||
boundingBox=objects; |
||||
return true; |
||||
}; |
||||
|
||||
} /* namespace cv */ |
||||
Loading…
Reference in new issue