crfmodel.h

Go to the documentation of this file.

#include <vector>
#include <stdlib.h>
#include <math.h>
#ifndef CRFMODEL_H
#define CRFMODEL_H

typedef std::vector<double> double_vector;
typedef std::vector< std::vector<double> > double_double_vector;

typedef std::vector<size_t> int_vector;
typedef std::vector< std::vector<size_t> > int_int_vector;

namespace CRF
{

    class CRFModel
    {
    public:
        CRFModel();

        CRFModel(int imgWidth, int imgHeight, int MAPSTEPS);

        virtual int_int_vector EstimateMAP(const double_double_vector Input);

        double EstimateLogLikelihood(const double_double_vector & Input, const int_int_vector & Output);

        virtual void EstimateModelParameters(const std::vector<double_double_vector> & InputSamples, const std::vector<int_int_vector> & OutputSamples, size_t Steps, double StepFactor);

    protected:
        double_vector weights;

        size_t imageWidth;

        size_t imageHeight;

        size_t paramCount;

        size_t statesCount;

        size_t MAPSteps;


        double EvaluatePartitionFunction(const double_double_vector & Input);

        double_vector EvaluateExpectedFeatureFunction(const double_double_vector & Input);

        double EvaluateOutputEnergy(const double_double_vector & Input, const int_int_vector & Output);

        double EvaluateNodeEnergy(const double_double_vector Input, const int_int_vector Output, size_t i, size_t j);

        double_vector EvaluateNodeVector(const double_double_vector Input, const int_int_vector Output, size_t i, size_t j);

        int_vector ConvertToNewNumberRepresentation(int number, int newRepresentationSymbolCount, int maxDigits);

    };


}

#endif // CRFMODEL_H