tracking.hpp

// This file is part of OpenCV project.
// It is subject to the license terms in the LICENSE file found in the top-level directory
// of this distribution and at http://opencv.org/license.html.
 
#ifndef OPENCV_CONTRIB_TRACKING_HPP
#define OPENCV_CONTRIB_TRACKING_HPP
 
#include "opencv2/core.hpp"
#include "opencv2/video/tracking.hpp"
 
namespace cv {
#ifndef CV_DOXYGEN
inline namespace tracking {
#endif
 
class CV_EXPORTS_W TrackerCSRT : public Tracker
{
protected:
    TrackerCSRT();  // use ::create()
public:
    virtual ~TrackerCSRT() CV_OVERRIDE;
 
    struct CV_EXPORTS_W_SIMPLE Params
    {
        CV_WRAP Params();
 
        CV_PROP_RW bool use_hog;
        CV_PROP_RW bool use_color_names;
        CV_PROP_RW bool use_gray;
        CV_PROP_RW bool use_rgb;
        CV_PROP_RW bool use_channel_weights;
        CV_PROP_RW bool use_segmentation;
 
        CV_PROP_RW std::string window_function; 
        CV_PROP_RW float kaiser_alpha;
        CV_PROP_RW float cheb_attenuation;
 
        CV_PROP_RW float template_size;
        CV_PROP_RW float gsl_sigma;
        CV_PROP_RW float hog_orientations;
        CV_PROP_RW float hog_clip;
        CV_PROP_RW float padding;
        CV_PROP_RW float filter_lr;
        CV_PROP_RW float weights_lr;
        CV_PROP_RW int num_hog_channels_used;
        CV_PROP_RW int admm_iterations;
        CV_PROP_RW int histogram_bins;
        CV_PROP_RW float histogram_lr;
        CV_PROP_RW int background_ratio;
        CV_PROP_RW int number_of_scales;
        CV_PROP_RW float scale_sigma_factor;
        CV_PROP_RW float scale_model_max_area;
        CV_PROP_RW float scale_lr;
        CV_PROP_RW float scale_step;
 
        CV_PROP_RW float psr_threshold; 
    };
 
    static CV_WRAP
    Ptr<TrackerCSRT> create(const TrackerCSRT::Params &parameters = TrackerCSRT::Params());
 
    //void init(InputArray image, const Rect& boundingBox) CV_OVERRIDE;
    //bool update(InputArray image, CV_OUT Rect& boundingBox) CV_OVERRIDE;
 
    CV_WRAP virtual void setInitialMask(InputArray mask) = 0;
};
 
 
class CV_EXPORTS_W TrackerKCF : public Tracker
{
protected:
    TrackerKCF();  // use ::create()
public:
    virtual ~TrackerKCF() CV_OVERRIDE;
 
    enum MODE {
      GRAY   = (1 << 0),
      CN     = (1 << 1),
      CUSTOM = (1 << 2)
    };
 
    struct CV_EXPORTS_W_SIMPLE Params
    {
        CV_WRAP Params();
 
        CV_PROP_RW float detect_thresh;         
        CV_PROP_RW float sigma;                 
        CV_PROP_RW float lambda;                
        CV_PROP_RW float interp_factor;         
        CV_PROP_RW float output_sigma_factor;   
        CV_PROP_RW float pca_learning_rate;     
        CV_PROP_RW bool resize;                  
        CV_PROP_RW bool split_coeff;             
        CV_PROP_RW bool wrap_kernel;             
        CV_PROP_RW bool compress_feature;        
        CV_PROP_RW int max_patch_size;           
        CV_PROP_RW int compressed_size;          
        CV_PROP_RW int desc_pca;        
        CV_PROP_RW int desc_npca;       
    };
 
    static CV_WRAP
    Ptr<TrackerKCF> create(const TrackerKCF::Params &parameters = TrackerKCF::Params());
 
    //void init(InputArray image, const Rect& boundingBox) CV_OVERRIDE;
    //bool update(InputArray image, CV_OUT Rect& boundingBox) CV_OVERRIDE;
 
    // FIXIT use interface
    typedef void (*FeatureExtractorCallbackFN)(const Mat, const Rect, Mat&);
    virtual void setFeatureExtractor(FeatureExtractorCallbackFN callback, bool pca_func = false) = 0;
};
 
 
 
#ifndef CV_DOXYGEN
}
#endif
}  // namespace
 
#endif // OPENCV_CONTRIB_TRACKING_HPP