RFHoG.h 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. /**
  2. * @file RFHoG.h
  3. * @brief computes the HoG features for each region
  4. * @author Björn Fröhlich and Alexander Arest
  5. * @date 02-10-2010
  6. */
  7. #ifndef RFHoGINCLUDE
  8. #define RFHoGINCLUDE
  9. #include <vislearning/nice.h>
  10. #include "core/basics/Config.h"
  11. #include "vislearning/cbaselib/LabeledSet.h"
  12. #include "vislearning/segmentation/RegionGraph.h"
  13. #include "vislearning/features/localfeatures/LocalFeature.h"
  14. #include "vislearning/features/regionfeatures/RegionFeatures.h"
  15. #include "vislearning/features/regionfeatures/RFStruct.h"
  16. namespace OBJREC {
  17. class RFHoG: public RFStruct
  18. {
  19. protected:
  20. /** @{ feature parameter */
  21. int amount_cell_x; // Amount of cells in x-Axis
  22. int amount_cell_y; // Amount of cells in y-Axis
  23. int block_x;
  24. int block_y;
  25. /** @} */
  26. public:
  27. /** simple constructor */
  28. RFHoG( const NICE::Config *conf);
  29. /** simple destructor */
  30. virtual ~RFHoG();
  31. /**
  32. * extracts the features for region rg (displayed in mask)
  33. * @param imgd_gradient allready computed gradientsmagnitudes
  34. * @param imgd_graddir allready computed gradientsdirection
  35. * @param rg input region graph
  36. * @param mask input region map
  37. * @param feats output features (in order of rg)
  38. */
  39. virtual void extract ( const NICE::FloatImage &imgd_gradient, const NICE::Image &imgd_graddir, const RegionGraph &rg, const NICE::Matrix &mask, NICE::VVector & feats );
  40. };
  41. } // namespace
  42. #endif