#include <ml.h>
Inherits CvStatModel.
Public Types | |
enum | { IDENTITY = 0, SIGMOID_SYM = 1, GAUSSIAN = 2 } |
enum | { UPDATE_WEIGHTS = 1, NO_INPUT_SCALE = 2, NO_OUTPUT_SCALE = 4 } |
Public Member Functions | |
CvANN_MLP () | |
CvANN_MLP (const CvMat *_layer_sizes, int _activ_func=SIGMOID_SYM, double _f_param1=0, double _f_param2=0) | |
virtual | ~CvANN_MLP () |
virtual void | create (const CvMat *_layer_sizes, int _activ_func=SIGMOID_SYM, double _f_param1=0, double _f_param2=0) |
virtual int | train (const CvMat *_inputs, const CvMat *_outputs, const CvMat *_sample_weights, const CvMat *_sample_idx=0, CvANN_MLP_TrainParams _params=CvANN_MLP_TrainParams(), int flags=0) |
virtual float | predict (const CvMat *_inputs, CvMat *_outputs) const |
CvANN_MLP (const cv::Mat &_layer_sizes, int _activ_func=SIGMOID_SYM, double _f_param1=0, double _f_param2=0) | |
virtual void | create (const cv::Mat &_layer_sizes, int _activ_func=SIGMOID_SYM, double _f_param1=0, double _f_param2=0) |
virtual int | train (const cv::Mat &_inputs, const cv::Mat &_outputs, const cv::Mat &_sample_weights, const cv::Mat &_sample_idx=cv::Mat(), CvANN_MLP_TrainParams _params=CvANN_MLP_TrainParams(), int flags=0) |
virtual float | predict (const cv::Mat &_inputs, cv::Mat &_outputs) const |
virtual void | clear () |
virtual void | read (CvFileStorage *fs, CvFileNode *node) |
virtual void | write (CvFileStorage *storage, const char *name) const |
int | get_layer_count () |
const CvMat * | get_layer_sizes () |
double * | get_weights (int layer) |
virtual void | save (const char *filename, const char *name=0) const |
virtual void | load (const char *filename, const char *name=0) |
Protected Member Functions | |
virtual bool | prepare_to_train (const CvMat *_inputs, const CvMat *_outputs, const CvMat *_sample_weights, const CvMat *_sample_idx, CvVectors *_ivecs, CvVectors *_ovecs, double **_sw, int _flags) |
virtual int | train_backprop (CvVectors _ivecs, CvVectors _ovecs, const double *_sw) |
virtual int | train_rprop (CvVectors _ivecs, CvVectors _ovecs, const double *_sw) |
virtual void | calc_activ_func (CvMat *xf, const double *bias) const |
virtual void | calc_activ_func_deriv (CvMat *xf, CvMat *deriv, const double *bias) const |
virtual void | set_activ_func (int _activ_func=SIGMOID_SYM, double _f_param1=0, double _f_param2=0) |
virtual void | init_weights () |
virtual void | scale_input (const CvMat *_src, CvMat *_dst) const |
virtual void | scale_output (const CvMat *_src, CvMat *_dst) const |
virtual void | calc_input_scale (const CvVectors *vecs, int flags) |
virtual void | calc_output_scale (const CvVectors *vecs, int flags) |
virtual void | write_params (CvFileStorage *fs) const |
virtual void | read_params (CvFileStorage *fs, CvFileNode *node) |
Protected Attributes | |
CvMat * | layer_sizes |
CvMat * | wbuf |
CvMat * | sample_weights |
double ** | weights |
double | f_param1 |
double | f_param2 |
double | min_val |
double | max_val |
double | min_val1 |
double | max_val1 |
int | activ_func |
int | max_count |
int | max_buf_sz |
CvANN_MLP_TrainParams | params |
CvRNG | rng |
const char * | default_model_name |
CvANN_MLP::CvANN_MLP | ( | ) |
CvANN_MLP::CvANN_MLP | ( | const CvMat * | _layer_sizes, |
int | _activ_func = SIGMOID_SYM , |
||
double | _f_param1 = 0 , |
||
double | _f_param2 = 0 |
||
) |
virtual CvANN_MLP::~CvANN_MLP | ( | ) | [virtual] |
CvANN_MLP::CvANN_MLP | ( | const cv::Mat & | _layer_sizes, |
int | _activ_func = SIGMOID_SYM , |
||
double | _f_param1 = 0 , |
||
double | _f_param2 = 0 |
||
) |
virtual void CvANN_MLP::create | ( | const CvMat * | _layer_sizes, |
int | _activ_func = SIGMOID_SYM , |
||
double | _f_param1 = 0 , |
||
double | _f_param2 = 0 |
||
) | [virtual] |
virtual int CvANN_MLP::train | ( | const CvMat * | _inputs, |
const CvMat * | _outputs, | ||
const CvMat * | _sample_weights, | ||
const CvMat * | _sample_idx = 0 , |
||
CvANN_MLP_TrainParams | _params = CvANN_MLP_TrainParams() , |
||
int | flags = 0 |
||
) | [virtual] |
virtual void CvANN_MLP::create | ( | const cv::Mat & | _layer_sizes, |
int | _activ_func = SIGMOID_SYM , |
||
double | _f_param1 = 0 , |
||
double | _f_param2 = 0 |
||
) | [virtual] |
virtual int CvANN_MLP::train | ( | const cv::Mat & | _inputs, |
const cv::Mat & | _outputs, | ||
const cv::Mat & | _sample_weights, | ||
const cv::Mat & | _sample_idx = cv::Mat() , |
||
CvANN_MLP_TrainParams | _params = CvANN_MLP_TrainParams() , |
||
int | flags = 0 |
||
) | [virtual] |
virtual void CvANN_MLP::clear | ( | ) | [virtual] |
Reimplemented from CvStatModel.
virtual void CvANN_MLP::read | ( | CvFileStorage * | fs, |
CvFileNode * | node | ||
) | [virtual] |
Reimplemented from CvStatModel.
virtual void CvANN_MLP::write | ( | CvFileStorage * | storage, |
const char * | name | ||
) | const [virtual] |
Reimplemented from CvStatModel.
int CvANN_MLP::get_layer_count | ( | ) |
const CvMat* CvANN_MLP::get_layer_sizes | ( | ) |
double* CvANN_MLP::get_weights | ( | int | layer ) |
virtual bool CvANN_MLP::prepare_to_train | ( | const CvMat * | _inputs, |
const CvMat * | _outputs, | ||
const CvMat * | _sample_weights, | ||
const CvMat * | _sample_idx, | ||
CvVectors * | _ivecs, | ||
CvVectors * | _ovecs, | ||
double ** | _sw, | ||
int | _flags | ||
) | [protected, virtual] |
virtual int CvANN_MLP::train_backprop | ( | CvVectors | _ivecs, |
CvVectors | _ovecs, | ||
const double * | _sw | ||
) | [protected, virtual] |
virtual int CvANN_MLP::train_rprop | ( | CvVectors | _ivecs, |
CvVectors | _ovecs, | ||
const double * | _sw | ||
) | [protected, virtual] |
virtual void CvANN_MLP::calc_activ_func | ( | CvMat * | xf, |
const double * | bias | ||
) | const [protected, virtual] |
virtual void CvANN_MLP::calc_activ_func_deriv | ( | CvMat * | xf, |
CvMat * | deriv, | ||
const double * | bias | ||
) | const [protected, virtual] |
virtual void CvANN_MLP::set_activ_func | ( | int | _activ_func = SIGMOID_SYM , |
double | _f_param1 = 0 , |
||
double | _f_param2 = 0 |
||
) | [protected, virtual] |
virtual void CvANN_MLP::init_weights | ( | ) | [protected, virtual] |
virtual void CvANN_MLP::scale_output | ( | const CvMat * | _src, |
CvMat * | _dst | ||
) | const [protected, virtual] |
virtual void CvANN_MLP::calc_input_scale | ( | const CvVectors * | vecs, |
int | flags | ||
) | [protected, virtual] |
virtual void CvANN_MLP::calc_output_scale | ( | const CvVectors * | vecs, |
int | flags | ||
) | [protected, virtual] |
virtual void CvANN_MLP::write_params | ( | CvFileStorage * | fs ) | const [protected, virtual] |
virtual void CvANN_MLP::read_params | ( | CvFileStorage * | fs, |
CvFileNode * | node | ||
) | [protected, virtual] |
virtual void CvStatModel::save | ( | const char * | filename, |
const char * | name = 0 |
||
) | const [virtual, inherited] |
virtual void CvStatModel::load | ( | const char * | filename, |
const char * | name = 0 |
||
) | [virtual, inherited] |
CvMat* CvANN_MLP::layer_sizes [protected] |
CvMat* CvANN_MLP::wbuf [protected] |
CvMat* CvANN_MLP::sample_weights [protected] |
double** CvANN_MLP::weights [protected] |
double CvANN_MLP::f_param1 [protected] |
double CvANN_MLP::f_param2 [protected] |
double CvANN_MLP::min_val [protected] |
double CvANN_MLP::max_val [protected] |
double CvANN_MLP::min_val1 [protected] |
double CvANN_MLP::max_val1 [protected] |
int CvANN_MLP::activ_func [protected] |
int CvANN_MLP::max_count [protected] |
int CvANN_MLP::max_buf_sz [protected] |
CvANN_MLP_TrainParams CvANN_MLP::params [protected] |
CvRNG CvANN_MLP::rng [protected] |
const char* CvStatModel::default_model_name [protected, inherited] |