@@ -978,8 +978,69 @@ class CV_EXPORTS_W TrackerVit : public Tracker
978
978
// bool update(InputArray image, CV_OUT Rect& boundingBox) CV_OVERRIDE;
979
979
};
980
980
981
+
982
+ /* * @brief the CSRT tracker
983
+
984
+ The implementation is based on @cite Lukezic_IJCV2018 Discriminative Correlation Filter with Channel and Spatial Reliability
985
+ */
986
+ class CV_EXPORTS_W TrackerCSRT : public Tracker
987
+ {
988
+ protected:
989
+ TrackerCSRT (); // use ::create()
990
+ public:
991
+ virtual ~TrackerCSRT () CV_OVERRIDE;
992
+
993
+ struct CV_EXPORTS_W_SIMPLE Params
994
+ {
995
+ CV_WRAP Params ();
996
+
997
+ CV_PROP_RW bool use_hog;
998
+ CV_PROP_RW bool use_color_names;
999
+ CV_PROP_RW bool use_gray;
1000
+ CV_PROP_RW bool use_rgb;
1001
+ CV_PROP_RW bool use_channel_weights;
1002
+ CV_PROP_RW bool use_segmentation;
1003
+
1004
+ CV_PROP_RW std::string window_function; // !< Window function: "hann", "cheb", "kaiser"
1005
+ CV_PROP_RW float kaiser_alpha;
1006
+ CV_PROP_RW float cheb_attenuation;
1007
+
1008
+ CV_PROP_RW float template_size;
1009
+ CV_PROP_RW float gsl_sigma;
1010
+ CV_PROP_RW float hog_orientations;
1011
+ CV_PROP_RW float hog_clip;
1012
+ CV_PROP_RW float padding;
1013
+ CV_PROP_RW float filter_lr;
1014
+ CV_PROP_RW float weights_lr;
1015
+ CV_PROP_RW int num_hog_channels_used;
1016
+ CV_PROP_RW int admm_iterations;
1017
+ CV_PROP_RW int histogram_bins;
1018
+ CV_PROP_RW float histogram_lr;
1019
+ CV_PROP_RW int background_ratio;
1020
+ CV_PROP_RW int number_of_scales;
1021
+ CV_PROP_RW float scale_sigma_factor;
1022
+ CV_PROP_RW float scale_model_max_area;
1023
+ CV_PROP_RW float scale_lr;
1024
+ CV_PROP_RW float scale_step;
1025
+
1026
+ CV_PROP_RW float psr_threshold; // !< we lost the target, if the psr is lower than this.
1027
+ };
1028
+
1029
+ /* * @brief Create CSRT tracker instance
1030
+ @param parameters CSRT parameters TrackerCSRT::Params
1031
+ */
1032
+ static CV_WRAP Ptr <TrackerCSRT> create (const TrackerCSRT::Params ¶meters = TrackerCSRT::Params());
1033
+
1034
+ CV_WRAP virtual void setInitialMask (InputArray mask) = 0;
1035
+
1036
+ // ! Return estimated tracking confidence
1037
+ CV_WRAP virtual float getTrackingScore () const = 0;
1038
+ };
1039
+
1040
+
981
1041
// ! @} video_track
982
1042
983
1043
} // cv
984
1044
1045
+
985
1046
#endif
0 commit comments