#include <fvutils/colormap/bayes/bayes_histos_to_lut.h>
Public Member Functions | |
BayesHistosToLut (std::map< hint_t, Histogram * > &histos, unsigned int d=1, hint_t fg_object=H_UNKNOWN, unsigned int w=256, unsigned int h=256) | |
Constructor. | |
~BayesHistosToLut () | |
Destructor. | |
std::string | getName () |
Get name. | |
float | getObjectProb (hint_t object) |
Get object probability. | |
float | getAPrioriProb (unsigned int u, unsigned int v, hint_t object) |
P(u, v| object). | |
float | getAPrioriProb (unsigned int y, unsigned int u, unsigned int v, hint_t object) |
P(u, v| object). | |
float | getAPosterioriProb (hint_t object, unsigned int u, unsigned int v) |
P(object| u, v). | |
float | getAPosterioriProb (hint_t object, unsigned int y, unsigned int u, unsigned int v) |
P(object| u, v). | |
hint_t | getMostLikelyObject (unsigned int u, unsigned int v) |
Get most likely object. | |
hint_t | getMostLikelyObject (unsigned int y, unsigned int u, unsigned int v) |
Get most likely object. | |
void | setMinProbability (float min_prob) |
Set min probability. | |
void | setMinProbForColor (float min_prob, hint_t hint) |
Set min probability for color. | |
YuvColormap * | get_colormap () |
Get generated color model. | |
void | calculateLutValues (bool penalty=false) |
Calculate LUT values. | |
void | calculateLutAllColors () |
Calculate all LUT colors. | |
void | saveLut (char *file) |
Save LUT to file. | |
void | save (std::string filename) |
Save LUT to file. |
Generates a YUV colormap.
Tim Niemueller
Daniel Beck
Definition at line 46 of file bayes_histos_to_lut.h.
firevision::BayesHistosToLut::BayesHistosToLut | ( | std::map< hint_t, Histogram * > & | histos, | |
unsigned int | d = 1 , |
|||
hint_t | object = H_UNKNOWN , |
|||
unsigned int | w = 256 , |
|||
unsigned int | h = 256 | |||
) |
Constructor.
histos | histograms | |
d | depth of lookup table | |
object | type of the foreground object | |
w | the width of the lookup table (u-resolution) | |
h | the height of the lookup table (v-resolution) |
Definition at line 65 of file bayes_histos_to_lut.cpp.
firevision::BayesHistosToLut::~BayesHistosToLut | ( | ) |
void firevision::BayesHistosToLut::calculateLutAllColors | ( | ) |
Calculate all LUT colors.
Definition at line 317 of file bayes_histos_to_lut.cpp.
References firevision::YuvColormap::deepness(), firevision::YuvColormap::depth(), firevision::ColorObjectMap::get_instance(), firevision::H_UNKNOWN, and firevision::YuvColormap::set().
void firevision::BayesHistosToLut::calculateLutValues | ( | bool | penalty = false |
) |
Calculate LUT values.
penalty | if true, non-ball colors are penalized |
Definition at line 384 of file bayes_histos_to_lut.cpp.
References firevision::YuvColormap::deepness(), firevision::YuvColormap::depth(), firevision::Histogram::get_average(), firevision::ColorObjectMap::get_instance(), firevision::Histogram::get_median(), firevision::Histogram::get_num_undos(), firevision::Histogram::get_value(), getMostLikelyObject(), firevision::H_BACKGROUND, firevision::H_BALL, firevision::H_FIELD, firevision::H_GOAL_BLUE, firevision::H_GOAL_YELLOW, firevision::H_LINE, firevision::H_ROBOT, firevision::H_ROBOT_OPP, firevision::H_UNKNOWN, firevision::YuvColormap::height(), firevision::YuvColormap::reset(), firevision::YuvColormap::set(), firevision::Histogram::set_value(), firevision::Histogram::switch_undo(), firevision::Histogram::undo(), and firevision::YuvColormap::width().
Referenced by firevision::BayesColormapGenerator::calc().
YuvColormap * firevision::BayesHistosToLut::get_colormap | ( | ) |
Get generated color model.
Definition at line 604 of file bayes_histos_to_lut.cpp.
Referenced by firevision::BayesColormapGenerator::BayesColormapGenerator(), and firevision::BayesColormapGenerator::load_histograms().
float firevision::BayesHistosToLut::getAPosterioriProb | ( | hint_t | object, | |
unsigned int | y, | |||
unsigned int | u, | |||
unsigned int | v | |||
) |
P(object| u, v).
Get a-posteriori probability.
object | objcet | |
y | YUV Y-value | |
u | YUV U-value | |
v | YUV V-value |
Definition at line 232 of file bayes_histos_to_lut.cpp.
References getAPrioriProb(), and getObjectProb().
float firevision::BayesHistosToLut::getAPosterioriProb | ( | hint_t | object, | |
unsigned int | u, | |||
unsigned int | v | |||
) |
P(object| u, v).
Get a-posteriori probability.
object | objcet | |
u | YUV U-value | |
v | YUV V-value |
Definition at line 203 of file bayes_histos_to_lut.cpp.
References getAPrioriProb(), and getObjectProb().
Referenced by getMostLikelyObject().
float firevision::BayesHistosToLut::getAPrioriProb | ( | unsigned int | y, | |
unsigned int | u, | |||
unsigned int | v, | |||
hint_t | object | |||
) |
P(u, v| object).
Get a-priori probability.
y | YUV Y-value | |
u | YUV U-value | |
v | YUV V-value | |
object | object. |
Definition at line 187 of file bayes_histos_to_lut.cpp.
float firevision::BayesHistosToLut::getAPrioriProb | ( | unsigned int | u, | |
unsigned int | v, | |||
hint_t | object | |||
) |
P(u, v| object).
Get a-priori probability.
u | YUV U-value | |
v | YUV V-value | |
object | object. |
Definition at line 166 of file bayes_histos_to_lut.cpp.
Referenced by getAPosterioriProb().
hint_t firevision::BayesHistosToLut::getMostLikelyObject | ( | unsigned int | y, | |
unsigned int | u, | |||
unsigned int | v | |||
) |
Get most likely object.
y | YUV Y-value | |
u | YUV U-value | |
v | YUV V-value |
Definition at line 291 of file bayes_histos_to_lut.cpp.
References getAPosterioriProb(), and firevision::H_UNKNOWN.
hint_t firevision::BayesHistosToLut::getMostLikelyObject | ( | unsigned int | u, | |
unsigned int | v | |||
) |
Get most likely object.
u | YUV U-value | |
v | YUV V-value |
Definition at line 259 of file bayes_histos_to_lut.cpp.
References getAPosterioriProb(), and firevision::H_UNKNOWN.
Referenced by calculateLutValues().
string firevision::BayesHistosToLut::getName | ( | ) |
float firevision::BayesHistosToLut::getObjectProb | ( | hint_t | object | ) |
Get object probability.
object | object |
Definition at line 108 of file bayes_histos_to_lut.cpp.
References firevision::H_BALL.
Referenced by getAPosterioriProb().
void firevision::BayesHistosToLut::save | ( | std::string | filename | ) |
Save LUT to file.
filename | file name |
Definition at line 550 of file bayes_histos_to_lut.cpp.
References firevision::ColormapFile::add_colormap(), and firevision::FireVisionDataFile::write().
void firevision::BayesHistosToLut::saveLut | ( | char * | file | ) |
Save LUT to file.
file | file name |
Definition at line 539 of file bayes_histos_to_lut.cpp.
References firevision::ColormapFile::add_colormap(), and firevision::FireVisionDataFile::write().
void firevision::BayesHistosToLut::setMinProbability | ( | float | min_prob | ) |
Set min probability.
min_prob | minimum probability |
Definition at line 562 of file bayes_histos_to_lut.cpp.
Referenced by firevision::BayesColormapGenerator::set_min_probability().
void firevision::BayesHistosToLut::setMinProbForColor | ( | float | min_prob, | |
hint_t | hint | |||
) |
Set min probability for color.
min_prob | minimum probability | |
hint | color hint |
Definition at line 573 of file bayes_histos_to_lut.cpp.
References firevision::H_BALL, firevision::H_FIELD, firevision::H_GOAL_BLUE, firevision::H_GOAL_YELLOW, firevision::H_LINE, and firevision::H_ROBOT.