SemSegLocal.h 990 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. /**
  2. * @file SemSegLocal.h
  3. * @brief semantic segmentation using image patches only
  4. * @author Erik Rodner
  5. * @date 05/08/2008
  6. */
  7. #ifndef SEMSEGLOCALINCLUDE
  8. #define SEMSEGLOCALINCLUDE
  9. #ifdef NOVISUAL
  10. #include <objrec/nice_nonvis.h>
  11. #else
  12. #include <objrec/nice.h>
  13. #endif
  14. #include "objrec/classifier/classifierbase/FeaturePoolClassifier.h"
  15. #include "SemanticSegmentation.h"
  16. namespace OBJREC {
  17. /** abstract interface for pixelwise localization systems */
  18. class SemSegLocal : public SemanticSegmentation
  19. {
  20. protected:
  21. bool save_cache;
  22. bool read_cache;
  23. std::string cache;
  24. FeaturePoolClassifier *fpc;
  25. public:
  26. /** simple constructor */
  27. SemSegLocal( const Config *conf, const MultiDataset *md );
  28. /** simple destructor */
  29. virtual ~SemSegLocal();
  30. void train ( const Config *conf, const MultiDataset *md );
  31. void semanticseg ( CachedExample *ce,
  32. NICE::Image & segresult,
  33. NICE::MultiChannelImageT<double> & probabilities );
  34. };
  35. } // namespace
  36. #endif