TrainEdgePottsCS.h

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

#include "TrainEdgePotts.h"

namespace DirectGraphicalModels
{
    enum ePotPenalApproach {
        eP_APP_PEN_CHAR,    
        eP_APP_PEN_PM,      
        eP_APP_PEN_EXP      
    }; 


    // ============================= Contrast-Sensitive Potts Train Class =============================
    class CTrainEdgePottsCS : public CTrainEdgePotts
    {
    public:
        DllExport CTrainEdgePottsCS(byte nStates, word nFeatures, ePotPenalApproach penApproach = eP_APP_PEN_EXP) 
            : CBaseRandomModel(nStates)
            , CTrainEdgePotts(nStates, nFeatures)
            , m_penApproach(penApproach) 
        {}
        DllExport virtual ~CTrainEdgePottsCS(void) {}


    protected:
        DllExport virtual Mat   calculateEdgePotentials(const Mat &featureVector1, const Mat &featureVector2, const vec_float_t &vParams) const;


    private:
        ePotPenalApproach   m_penApproach;
    };
}