MeanShiftTracker.h

Go to the documentation of this file.

/********************************************************************
**                Image Component Library (ICL)                    **
**                                                                 **
** Copyright (C) 2006-2013 CITEC, University of Bielefeld          **
**                         Neuroinformatics Group                  **
** Website: www.iclcv.org and                                      **
**          http://opensource.cit-ec.de/projects/icl               **
**                                                                 **
** File   : ICLCV/src/ICLCV/MeanShiftTracker.h                     **
** Module : ICLCV                                                  **
** Authors: Christof Elbrechter                                    **
**                                                                 **
**                                                                 **
** GNU LESSER GENERAL PUBLIC LICENSE                               **
** This file may be used under the terms of the GNU Lesser General **
** Public License version 3.0 as published by the                  **
**                                                                 **
** Free Software Foundation and appearing in the file LICENSE.GPL  **
** included in the packaging of this file.  Please review the      **
** following information to ensure the license requirements will   **
** be met: http://www.gnu.org/licenses/lgpl-3.0.txt                **
**                                                                 **
** The development of this software was supported by the           **
** Excellence Cluster EXC 277 Cognitive Interaction Technology.    **
** The Excellence Cluster EXC 277 is a grant of the Deutsche       **
** Forschungsgemeinschaft (DFG) in the context of the German       **
** Excellence Initiative.                                          **
**                                                                 **
********************************************************************/

#pragma once

#include <ICLCore/Img.h>

namespace icl {
  namespace cv{
  
    class MeanShiftTracker {
      
      public:
      
      enum kernelType {
        epanechnikov,
        gauss
      }; 
      private:
      
      
      int m_bandwidth;
      
      kernelType m_kernelType;
  

      core::Img32f m_kernelImage;
      

      utils::Point32f applyMeanShiftStep(const core::Img32f &image, const utils::Point32f &pos);
  
      public:
  

      static core::Img32f generateEpanechnikov(int bandwidth);
  

      static core::Img32f generateGauss(int bandwidth,float stdDev);
  

      MeanShiftTracker(kernelType type, int bandwidth, float stdDev=1);
  
      

      void setKernel(kernelType type, int bandwidth, float stdDev=1);
  
      int getKernel() const { return m_kernelType;  }
      
      const core::Img32f &getKernelImage() const { return m_kernelImage; }
      
      int getBandwidth() const { return m_bandwidth; }
      

      const utils::Point32f step(const core::Img32f &weigthImage, const utils::Point32f &initialPoint,  int maxCycles=-1, float convergenceCriterion=1.0, bool *converged=0);
  
    };
  } // namespace cv
}