NICE_SemSeg/semseg/postsegmentation/PSSLocalizationPrior.h

/** 
* @file PSSLocalizationPrior.h
* @brief incorporate prior from localization results
* @author Erik Rodner
* @date 03/19/2009

*/
#ifndef PSSLOCALIZATIONPRIORINCLUDE
#define PSSLOCALIZATIONPRIORINCLUDE

#ifdef NOVISUAL
#include <objrec/nice_nonvis.h>
#else
#include <objrec/nice.h>
#endif
 
#include "objrec/cbaselib/LocalizationResult.h"
#include "PostSemSeg.h"


namespace OBJREC {

/** incorporate prior from localization results */
class PSSLocalizationPrior : public PostSemSeg
{

    protected:
	int subsamplex;
	int subsampley;
	double alphaDetectionPrior;
	map<string, LocalizationResult *> detresults;


	// refactor-nice.pl: check this substitution
	// old: void loadDetectionResults ( const string & dir,
	void loadDetectionResults ( const std::string & dir,
			       map<string, LocalizationResult *> & results,
			       const ClassNames *classNames );

    public:
  
	/** simple constructor */
	// refactor-nice.pl: check this substitution
	// old: PSSLocalizationPrior( const string & detectiondir, const ClassNames *classNames,
	PSSLocalizationPrior( const std::string & detectiondir, const ClassNames *classNames,
			      double alphaDetectionPrior,  int subsamplex = 1, int subsampley = 1 );
      
	/** simple destructor */
	virtual ~PSSLocalizationPrior();
	
	// refactor-nice.pl: check this substitution
	// old: virtual void postprocess ( Image & result, GenericImage<double> & probabilities );
	virtual void postprocess ( NICE::Image & result, GenericImage<double> & probabilities );


     
};


} // namespace

#endif