30 #ifndef _FIREVISION_COLORMODEL_BAYES_HISTOS_TO_LUT_H_
31 #define _FIREVISION_COLORMODEL_BAYES_HISTOS_TO_LUT_H_
33 #include <fvutils/base/roi.h>
38 namespace firevision {
48 hint_t fg_object = H_UNKNOWN,
50 unsigned int h = 256);
57 float getAPrioriProb(
unsigned int u,
unsigned int v, hint_t
object);
58 float getAPrioriProb(
unsigned int y,
unsigned int u,
unsigned int v, hint_t
object);
61 float getAPosterioriProb(hint_t
object,
unsigned int y,
unsigned int u,
unsigned int v);
79 void save(std::string filename);
82 std::map<hint_t, Histogram *> &histograms;
83 std::map<hint_t, unsigned int> numberOfOccurrences;
84 std::map<hint_t, float> object_probabilities;
93 float min_probability;
98 float min_prob_yellow;
100 float min_prob_white;
101 float min_prob_black;
LUT generation by using Bayesian method on histograms.
float getObjectProb(hint_t object)
Get object probability.
void setMinProbForColor(float min_prob, hint_t hint)
Set min probability for color.
YuvColormap * get_colormap()
Get generated color model.
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.
std::string getName()
Get name.
void calculateLutValues(bool penalty=false)
Calculate LUT values.
void saveLut(char *file)
Save LUT to file.
~BayesHistosToLut()
Destructor.
void save(std::string filename)
Save LUT to file.
void calculateLutAllColors()
Calculate all LUT colors.
float getAPosterioriProb(hint_t object, unsigned int u, unsigned int v)
P(object| u, v).
float getAPrioriProb(unsigned int u, unsigned int v, hint_t object)
P(u, v| object).
hint_t getMostLikelyObject(unsigned int u, unsigned int v)
Get most likely object.
void setMinProbability(float min_prob)
Set min probability.