onlineMIL.hpp 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118
  1. /*M///////////////////////////////////////////////////////////////////////////////////////
  2. //
  3. // IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
  4. //
  5. // By downloading, copying, installing or using the software you agree to this license.
  6. // If you do not agree to this license, do not download, install,
  7. // copy or use the software.
  8. //
  9. //
  10. // License Agreement
  11. // For Open Source Computer Vision Library
  12. //
  13. // Copyright (C) 2013, OpenCV Foundation, all rights reserved.
  14. // Third party copyrights are property of their respective owners.
  15. //
  16. // Redistribution and use in source and binary forms, with or without modification,
  17. // are permitted provided that the following conditions are met:
  18. //
  19. // * Redistribution's of source code must retain the above copyright notice,
  20. // this list of conditions and the following disclaimer.
  21. //
  22. // * Redistribution's in binary form must reproduce the above copyright notice,
  23. // this list of conditions and the following disclaimer in the documentation
  24. // and/or other materials provided with the distribution.
  25. //
  26. // * The name of the copyright holders may not be used to endorse or promote products
  27. // derived from this software without specific prior written permission.
  28. //
  29. // This software is provided by the copyright holders and contributors "as is" and
  30. // any express or implied warranties, including, but not limited to, the implied
  31. // warranties of merchantability and fitness for a particular purpose are disclaimed.
  32. // In no event shall the Intel Corporation or contributors be liable for any direct,
  33. // indirect, incidental, special, exemplary, or consequential damages
  34. // (including, but not limited to, procurement of substitute goods or services;
  35. // loss of use, data, or profits; or business interruption) however caused
  36. // and on any theory of liability, whether in contract, strict liability,
  37. // or tort (including negligence or otherwise) arising in any way out of
  38. // the use of this software, even if advised of the possibility of such damage.
  39. //
  40. //M*/
  41. #ifndef __OPENCV_ONLINEMIL_HPP__
  42. #define __OPENCV_ONLINEMIL_HPP__
  43. #include "opencv2/core.hpp"
  44. #include <limits>
  45. namespace cv
  46. {
  47. //! @addtogroup tracking
  48. //! @{
  49. //TODO based on the original implementation
  50. //http://vision.ucsd.edu/~bbabenko/project_miltrack.shtml
  51. class ClfOnlineStump;
  52. class CV_EXPORTS ClfMilBoost
  53. {
  54. public:
  55. struct CV_EXPORTS Params
  56. {
  57. Params();
  58. int _numSel;
  59. int _numFeat;
  60. float _lRate;
  61. };
  62. ClfMilBoost();
  63. ~ClfMilBoost();
  64. void init( const ClfMilBoost::Params &parameters = ClfMilBoost::Params() );
  65. void update( const Mat& posx, const Mat& negx );
  66. std::vector<float> classify( const Mat& x, bool logR = true );
  67. inline float sigmoid( float x )
  68. {
  69. return 1.0f / ( 1.0f + exp( -x ) );
  70. }
  71. private:
  72. uint _numsamples;
  73. ClfMilBoost::Params _myParams;
  74. std::vector<int> _selectors;
  75. std::vector<ClfOnlineStump*> _weakclf;
  76. uint _counter;
  77. };
  78. class ClfOnlineStump
  79. {
  80. public:
  81. float _mu0, _mu1, _sig0, _sig1;
  82. float _q;
  83. int _s;
  84. float _log_n1, _log_n0;
  85. float _e1, _e0;
  86. float _lRate;
  87. ClfOnlineStump();
  88. ClfOnlineStump( int ind );
  89. void init();
  90. void update( const Mat& posx, const Mat& negx, const cv::Mat_<float> & posw = cv::Mat_<float>(), const cv::Mat_<float> & negw = cv::Mat_<float>() );
  91. bool classify( const Mat& x, int i );
  92. float classifyF( const Mat& x, int i );
  93. std::vector<float> classifySetF( const Mat& x );
  94. private:
  95. bool _trained;
  96. int _ind;
  97. };
  98. //! @}
  99. } /* namespace cv */
  100. #endif