NICE_SemSeg/semseg/SemSegTools.h

/** 
* @file SemSegTools.h
* @brief tools for semantic segmentation
* @author Erik Rodner
* @date 03/19/2009

*/
#ifndef SEMSEGTOOLSINCLUDE
#define SEMSEGTOOLSINCLUDE

#ifdef NOVISUAL
#include <objrec/nice_nonvis.h>
#else
#include <objrec/nice.h>
#endif
  
#include "objrec/baselib/Config.h"
#include "objrec/cbaselib/MultiDataset.h"
#include "objrec/cbaselib/Example.h"
#include "objrec/cbaselib/CachedExample.h"


namespace OBJREC {

/** tools for semantic segmentation */
class SemSegTools
{

    protected:

    public:
 
	/** collect pixel-wise training examples 
	    from a set of images 
	    @param conf includes settings about grid size etc.
	    @param section section of the config
	    @param train set of training images with localization information
	    @param cn classNames object
	    @param examples resulting pixel-wise examples
	    @param imgexamples image based caching structure referenced by pixel-wise examples
	*/
	static void collectTrainingExamples ( 
			  const Config * conf,
			  const std::string & section,
			  const LabeledSet & train,
			  const ClassNames & cn,
			  Examples & examples,
			  vector<CachedExample *> & imgexamples );
    
};


} // namespace

#endif