GMHIKernelRaw.h 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. /**
  2. * @file GMHIKernelRaw.h
  3. * @author Erik Rodner, Alexander Freytag
  4. * @brief Fast multiplication with histogram intersection kernel matrices (Interface)
  5. */
  6. #ifndef _NICE_GMHIKERNELRAWINCLUDE
  7. #define _NICE_GMHIKERNELRAWINCLUDE
  8. #include <vector>
  9. #include <core/algebra/GenericMatrix.h>
  10. namespace NICE {
  11. /**
  12. * @class GMHIKernel
  13. * @brief Fast multiplication with histogram intersection kernel matrices
  14. * @author Erik Rodner, Alexander Freytag
  15. */
  16. class GMHIKernelRaw : public GenericMatrix
  17. {
  18. protected:
  19. typedef struct sparseVectorElement {
  20. uint example_index;
  21. double value;
  22. bool operator< (const sparseVectorElement & a) const
  23. {
  24. return value < a.value;
  25. }
  26. } sparseVectorElement;
  27. sparseVectorElement **examples_raw;
  28. double **table_A;
  29. double **table_B;
  30. uint *nnz_per_dimension;
  31. uint num_dimension;
  32. uint num_examples;
  33. double d_noise;
  34. void initData ( const std::vector< const NICE::SparseVector *> & examples );
  35. void cleanupData ();
  36. public:
  37. /** simple constructor */
  38. GMHIKernelRaw( const std::vector< const NICE::SparseVector *> & examples, const double d_noise = 0.1 );
  39. /** multiply with a vector: A*x = y; this is not really const anymore!! */
  40. virtual void multiply (NICE::Vector & y, const NICE::Vector & x) const;
  41. /** get the number of rows in A */
  42. virtual uint rows () const;
  43. /** get the number of columns in A */
  44. virtual uint cols () const;
  45. /** simple destructor */
  46. virtual ~GMHIKernelRaw();
  47. };
  48. }
  49. #endif