TrainEdgePotts.h

// Potts training model for pairwise potentials
// Written by Sergey G. Kosov in 2015 for Project X
#pragma once

#include "TrainEdge.h"

namespace DirectGraphicalModels
{
    // ============================= Potts Train Class =============================
    class CTrainEdgePotts : public CTrainEdge
    {
    public:
        DllExport CTrainEdgePotts(byte nStates, word nFeatures) : CBaseRandomModel(nStates), CTrainEdge(nStates, nFeatures) {}
        DllExport virtual ~CTrainEdgePotts(void) {}

        DllExport virtual void  reset(void) {}  

        DllExport virtual void  addFeatureVecs(const Mat &featureVector1, byte gt1, const Mat &featureVector2, byte gt2) {}


    protected:
        DllExport virtual void  saveFile(FILE *pFile) const {}
        DllExport virtual void  loadFile(FILE *pFile) {} 
        DllExport virtual Mat   calculateEdgePotentials(const Mat &featureVector1, const Mat &featureVector2, const vec_float_t &vParams) const;
    };
}