NICE_SemSeg/semseg/postsegmentation/PSSImageLevelPrior.h

/** 
* @file PSSImageLevelPrior.h
* @brief incorporate prior from image categorization method
* @author Erik Rodner
* @date 03/19/2009

*/
#ifndef PSSIMAGELEVELPRIORINCLUDE
#define PSSIMAGELEVELPRIORINCLUDE

#ifdef NOVISUAL
#include <objrec/nice_nonvis.h>
#else
#include <objrec/nice.h>
#endif
  
#include "PostSemSeg.h"
#include "objrec/math/mathbase/FullVector.h"


namespace OBJREC {

/** incorporate prior from image categorization method */
class PSSImageLevelPrior : public PostSemSeg
{

    protected:
	int imagePriorMethod;
	FullVector prior;

	int priorK;
	double alphaImagePrior;

    public:
	
	enum {
	    IMAGE_PRIOR_BEST_K = 0,
	    IMAGE_PRIOR_PSEUDOPROB
	};
  
	/** simple constructor */
	PSSImageLevelPrior( int imagePriorMethod, int priorK, double alphaImagePrior );
      
	/** simple destructor */
	virtual ~PSSImageLevelPrior();
	
	void setPrior ( FullVector & prior );

	// refactor-nice.pl: check this substitution
	// old: void postprocess ( Image & result, GenericImage<double> & probabilities );
	void postprocess ( NICE::Image & result, GenericImage<double> & probabilities );
     
};


} // namespace

#endif