tracker.cpp 3.37 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43
/*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"

44 45 46 47 48 49
#undef BOILERPLATE_CODE
#define BOILERPLATE_CODE(name,classname)\
  if(trackerType==name){\
      return classname::createTracker();\
  }

50 51 52 53 54 55 56 57 58 59 60
namespace cv
{

/*
 *  Tracker
 */

Tracker::~Tracker()
{
}

61
bool Tracker::init( const Mat& image, const Rect2d& boundingBox )
62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80
{

  if( isInit )
  {
    return false;
  }

  if( image.empty() )
    return false;

  sampler = Ptr<TrackerSampler>( new TrackerSampler() );
  featureSet = Ptr<TrackerFeatureSet>( new TrackerFeatureSet() );
  model = Ptr<TrackerModel>();

  bool initTracker = initImpl( image, boundingBox );

  //check if the model component is initialized
  if( model == 0 )
  {
81
    CV_Error( -1, "The model is not initialized" );
82 83 84 85 86 87 88 89 90 91 92
    return false;
  }

  if( initTracker )
  {
    isInit = true;
  }

  return initTracker;
}

93
bool Tracker::update( const Mat& image, Rect2d& boundingBox )
94 95 96 97 98 99 100 101 102 103 104 105 106 107 108
{

  if( !isInit )
  {
    return false;
  }

  if( image.empty() )
    return false;

  return updateImpl( image, boundingBox );
}

Ptr<Tracker> Tracker::create( const String& trackerType )
{
109 110
  BOILERPLATE_CODE("MIL",TrackerMIL);
  BOILERPLATE_CODE("BOOSTING",TrackerBoosting);
Alex Leontiev's avatar
Alex Leontiev committed
111 112
  BOILERPLATE_CODE("MEDIANFLOW",TrackerMedianFlow);
  BOILERPLATE_CODE("TLD",TrackerTLD);
113
  BOILERPLATE_CODE("KCF",TrackerKCF);
114
  return Ptr<Tracker>();
115 116 117
}

} /* namespace cv */