curfil  ..
 All Classes Functions Variables Typedefs Friends Groups Pages
Public Member Functions | List of all members
curfil::RandomTreeImage Class Reference

A random tree in a random forest, for RGB-D images. More...

#include <random_tree_image.h>

Public Member Functions

 RandomTreeImage (int id, const TrainingConfiguration &configuration)
 RandomTreeImage (boost::shared_ptr< RandomTree< PixelInstance, ImageFeatureFunction > > tree, const TrainingConfiguration &configuration, const cuv::ndarray< WeightType, cuv::host_memory_space > &classLabelPriorDistribution)
void train (const std::vector< LabeledRGBDImage > &trainLabelImages, RandomSource &randomSource, size_t subsampleCount, size_t numLabels)
 Trains the tree from the given training images.
void test (const RGBDImage *image, LabelImage &prediction) const
 Predicts the given image and outputs the prediction.
void normalizeHistograms (const double histogramBias)
 Normalize the leaf-node histograms in the tree.
const boost::shared_ptr
< RandomTree< PixelInstance,
ImageFeatureFunction > > & 
getTree () const
const cuv::ndarray< WeightType,
cuv::host_memory_space > & 
getClassLabelPriorDistribution () const
size_t getId () const
bool shouldIgnoreLabel (const LabelType &label) const

Detailed Description

A random tree in a random forest, for RGB-D images.

Definition at line 1075 of file random_tree_image.h.

Constructor & Destructor Documentation

curfil::RandomTreeImage::RandomTreeImage ( int  id,
const TrainingConfiguration configuration 
)
Parameters
idtree number
configurationtraining configuration of the tree
curfil::RandomTreeImage::RandomTreeImage ( boost::shared_ptr< RandomTree< PixelInstance, ImageFeatureFunction > >  tree,
const TrainingConfiguration configuration,
const cuv::ndarray< WeightType, cuv::host_memory_space > &  classLabelPriorDistribution 
)
Parameters
tree
configurationtraining configuration of the tree
classLabelPriorDistributionprior distributions of the classes

Member Function Documentation

const cuv::ndarray<WeightType, cuv::host_memory_space>& curfil::RandomTreeImage::getClassLabelPriorDistribution ( ) const
inline
Returns
prior distribution of classes

Definition at line 1121 of file random_tree_image.h.

size_t curfil::RandomTreeImage::getId ( ) const
inline
Returns
tree id

Definition at line 1128 of file random_tree_image.h.

const boost::shared_ptr<RandomTree<PixelInstance, ImageFeatureFunction> >& curfil::RandomTreeImage::getTree ( ) const
inline
Returns
the underlying tree

Definition at line 1114 of file random_tree_image.h.

void curfil::RandomTreeImage::normalizeHistograms ( const double  histogramBias)

Normalize the leaf-node histograms in the tree.

Parameters
histogramBiasthe histogram bias for the normalization
bool curfil::RandomTreeImage::shouldIgnoreLabel ( const LabelType &  label) const
Returns
whether the given label is one that the user specified as being ignored

The documentation for this class was generated from the following file: