Colormap Generator using Bayes method. More...
#include <>>
Public Member Functions | |
BayesColormapGenerator (unsigned int lut_depth=1, hint_t fg_object=H_UNKNOWN, unsigned int lut_width=256, unsigned int lut_height=256) | |
Constructor. | |
~BayesColormapGenerator () | |
Destructor. | |
virtual void | set_fg_object (hint_t object) |
Set foreground object. | |
virtual void | set_buffer (unsigned char *buffer, unsigned int width, unsigned int height) |
Set buffer. | |
virtual YuvColormap * | get_current () |
Get current color model. | |
virtual void | consider () |
Consider current image. | |
virtual void | calc () |
Calculate. | |
virtual void | undo () |
Undo last inclusion. | |
virtual void | reset () |
Reset color model. | |
virtual void | reset_undo () |
Reset undo. | |
virtual void | set_selection (std::vector< fawkes::rectangle_t > region) |
Set selection. | |
virtual bool | has_histograms () |
Check if this color model uses histograms. | |
virtual std::map< hint_t, Histogram * > * | get_histograms () |
Get histograms. | |
virtual void | load_histograms (const char *filename) |
Load histogram from a file. | |
virtual void | save_histograms (const char *filename) |
Save histograms to a file. | |
void | set_min_probability (float min_prob) |
Set min probability. |
Colormap Generator using Bayes method.
Definition at line 40 of file bayes_generator.h.
firevision::BayesColormapGenerator::BayesColormapGenerator | ( | unsigned int | lut_depth = 1 , |
|
hint_t | fg_object = H_UNKNOWN , |
|||
unsigned int | lut_width = 256 , |
|||
unsigned int | lut_height = 256 | |||
) |
Constructor.
lut_depth | the depth of the lookup table | |
fg_object | the type of a foreground object | |
lut_width | the width of the lookup table (u-resolution) | |
lut_height | the height of the lookup table (v-resolution) |
Definition at line 57 of file bayes_generator.cpp.
References firevision::BayesHistosToLut::get_colormap(), and set_fg_object().
firevision::BayesColormapGenerator::~BayesColormapGenerator | ( | ) |
Destructor.
Definition at line 78 of file bayes_generator.cpp.
void firevision::BayesColormapGenerator::calc | ( | ) | [virtual] |
Calculate.
Implements firevision::ColormapGenerator.
Definition at line 246 of file bayes_generator.cpp.
References firevision::BayesHistosToLut::calculateLutValues().
Referenced by ColorTrainWidget::add_to_colormap(), ColorTrainWidget::load_histograms(), and load_histograms().
void firevision::BayesColormapGenerator::consider | ( | ) | [virtual] |
Consider current image.
Implements firevision::ColormapGenerator.
Definition at line 201 of file bayes_generator.cpp.
Referenced by ColorTrainWidget::add_to_colormap().
YuvColormap * firevision::BayesColormapGenerator::get_current | ( | ) | [virtual] |
Get current color model.
Implements firevision::ColormapGenerator.
Definition at line 142 of file bayes_generator.cpp.
Referenced by ColorTrainWidget::add_to_colormap(), ColorTrainWidget::draw_segmentation_result(), ColorTrainWidget::get_colormap(), ColorTrainWidget::load_colormap(), ColorTrainWidget::load_histograms(), and ColorTrainWidget::save_colormap().
std::map< hint_t, Histogram * > * firevision::BayesColormapGenerator::get_histograms | ( | ) | [virtual] |
Get histograms.
Implements firevision::ColormapGenerator.
Definition at line 327 of file bayes_generator.cpp.
bool firevision::BayesColormapGenerator::has_histograms | ( | ) | [virtual] |
Check if this color model uses histograms.
Implements firevision::ColormapGenerator.
Definition at line 317 of file bayes_generator.cpp.
void firevision::BayesColormapGenerator::load_histograms | ( | const char * | filename | ) | [virtual] |
Load histogram from a file.
filename | the filename |
Definition at line 337 of file bayes_generator.cpp.
References calc(), firevision::HistogramBlock::depth(), firevision::BayesHistosToLut::get_colormap(), firevision::HistogramBlock::height(), firevision::HistogramFile::histogram_blocks(), firevision::FireVisionDataFile::read(), firevision::FireVisionDataFile::set_owns_blocks(), and firevision::HistogramBlock::width().
Referenced by ColorTrainWidget::load_histograms().
void firevision::BayesColormapGenerator::reset | ( | ) | [virtual] |
Reset color model.
Implements firevision::ColormapGenerator.
Definition at line 273 of file bayes_generator.cpp.
References firevision::YuvColormap::reset().
Referenced by ColorTrainWidget::reset_colormap().
void firevision::BayesColormapGenerator::reset_undo | ( | ) | [virtual] |
Reset undo.
Implements firevision::ColormapGenerator.
Definition at line 297 of file bayes_generator.cpp.
Referenced by ColorTrainWidget::add_to_colormap().
void firevision::BayesColormapGenerator::save_histograms | ( | const char * | filename | ) | [virtual] |
Save histograms to a file.
filename | the filename |
Definition at line 448 of file bayes_generator.cpp.
References firevision::HistogramFile::add_histogram_block(), firevision::HistogramBlock::set_object_type(), firevision::FireVisionDataFile::set_owns_blocks(), and firevision::FireVisionDataFile::write().
Referenced by ColorTrainWidget::save_histograms().
void firevision::BayesColormapGenerator::set_buffer | ( | unsigned char * | buffer, | |
unsigned int | width, | |||
unsigned int | height | |||
) | [virtual] |
Set buffer.
buffer | image buffer | |
width | image width | |
height | image height |
Implements firevision::ColormapGenerator.
Definition at line 121 of file bayes_generator.cpp.
Referenced by ColorTrainWidget::add_to_colormap().
void firevision::BayesColormapGenerator::set_fg_object | ( | hint_t | object | ) | [virtual] |
Set foreground object.
object | the new foreground object |
Definition at line 100 of file bayes_generator.cpp.
Referenced by ColorTrainWidget::add_to_colormap(), and BayesColormapGenerator().
void firevision::BayesColormapGenerator::set_min_probability | ( | float | min_prob | ) |
Set min probability.
min_prob | min probability. |
Definition at line 193 of file bayes_generator.cpp.
References firevision::BayesHistosToLut::setMinProbability().
void firevision::BayesColormapGenerator::set_selection | ( | std::vector< fawkes::rectangle_t > | region | ) | [virtual] |
Set selection.
region | selected region. |
Definition at line 164 of file bayes_generator.cpp.
Referenced by ColorTrainWidget::add_to_colormap().
void firevision::BayesColormapGenerator::undo | ( | ) | [virtual] |
Undo last inclusion.
Implements firevision::ColormapGenerator.
Definition at line 255 of file bayes_generator.cpp.