NICE_SemSeg/semseg/SemSegSTF.h

/** 
* @file SemSegSTF.h
* @brief Localization system
* @author Erik Rodner
* @date 02/11/2008

*/
#ifndef SemSegSTFINCLUDE
#define SemSegSTFINCLUDE

#include "SemanticSegmentation.h"
#include "vislearning/classifier/fpclassifier/randomforest/DecisionNode.h"
#include "vislearning/classifier/fpclassifier/randomforest/FPCRandomForests.h"

namespace OBJREC {

/** Localization system */
class SemSegSTF : public SemanticSegmentation
{
    protected:
 
	int numClasses;
	int maxdepthSegmentationForest;
	int maxdepthSegmentationForestScores;

	bool use_semantic_features;
	bool use_pixelpair_features;

	int subsamplex;
	int subsampley;

	bool read_seg_cache;
	bool read_pixel_cache;

	std::string cacheseg;

	std::string cachepixel;

	std::string resultsdatadir;

	double alphaDetectionPrior;
	double alphaImagePrior;
	
	int imagePriorMethod;
	int priorK;

	int detectionPriorMethod;

	/** 
	    assign all pixels with
	    normalized entropy $E/log(n)$ above
	    this threshold to the
	    background class 
	*/
	double entropyThreshold;

	int backgroundModelType;

	std::map<DecisionNode *, std::pair<long, int> > index;
	FPCRandomForests *fpcSeg;
	FPCRandomForests *fpcPixel;

	std::map<std::string, LocalizationResult *> detresults;

	void fillCachePixel ( CachedExample *ce );
	void fillCacheSegmentation ( CachedExample *ce );

    public:
	/** simple constructor */
	SemSegSTF( const Config *conf,
  	              const MultiDataset *md );
      
	/** simple destructor */
	virtual ~SemSegSTF();

	void semanticseg ( CachedExample *ce, 
			   NICE::Image & segresult,
			   NICE::MultiChannelImageT<double> & probabilities );
};
} // namespace

#endif