drwnPixelSegCRFInference.h

/*****************************************************************************
** DARWIN: A FRAMEWORK FOR MACHINE LEARNING RESEARCH AND DEVELOPMENT
** Distributed under the terms of the BSD license (see the LICENSE file)
** Copyright (c) 2007-2015, Stephen Gould
** All rights reserved.
**
******************************************************************************
** FILENAME:    drwnPixelSegCRFInference.h
** AUTHOR(S):   Stephen Gould <stephen.gould@anu.edu.au>
**
*****************************************************************************/

#pragma once

#include <cstdlib>
#include <cassert>
#include <iostream>
#include <fstream>
#include <iomanip>
#include <vector>

#include "Eigen/Core"

#include "cv.h"

#include "drwnBase.h"
#include "drwnPGM.h"
#include "drwnVision.h"

using namespace std;

// drwnPixelSegCRFInference class -----------------------------------------

class drwnPixelSegCRFInference {
 public:
    drwnPixelSegCRFInference() { /* do nothing */ };
    virtual ~drwnPixelSegCRFInference() { /* do nothing */ };

    void alphaExpansion(drwnSegImageInstance *instance, double contrastStrength = 0.0,
        double higherOrderStrength = 0.0, double longRangeStrength = 0.0) const;

    double energy(const drwnSegImageInstance *instance, double contrastStrength = 0.0,
        double higherOrderStrength = 0.0, double longRangeStrength = 0.0) const;

 protected:
    virtual void addAuxiliaryVariables(drwnMaxFlow *g, const drwnSegImageInstance *instance) const;

    double addUnaryTerms(drwnMaxFlow *g, const drwnSegImageInstance *instance,
        int alpha, int varOffset = 0) const;

    double addPairwiseContrastTerms(drwnMaxFlow *g, double contrastStrength,
        const drwnSegImageInstance *instance, int alpha, int varOffset = 0) const;

    double addLongRangePairwiseTerms(drwnMaxFlow *g, double longRangeStrength,
        const drwnSegImageInstance *instance, int alpha, int varOffset = 0) const;

    virtual double addHigherOrderTerms(drwnMaxFlow *g, double higherOrderStrength,
        const drwnSegImageInstance* instance, int alpha, int varOffset = 0) const;

    double computeUnaryEnergy(const drwnSegImageInstance *instance) const;

    double computePairwiseContrastEnergy(const drwnSegImageInstance *instance, double contrastStrength) const;

    double computeLongRangePairwiseEnergy(const drwnSegImageInstance *instance, double longRangeStrength) const;

    virtual double computeHigherOrderEnergy(const drwnSegImageInstance *instance,
        double higherOrderStrength) const;
};

// drwnRobustPottsCRFInference -----------------------------------------------

class drwnRobustPottsCRFInference : public drwnPixelSegCRFInference {
 public:
    double _eta;               
    bool _bWeightBySize;       

 public:
    drwnRobustPottsCRFInference(double eta = 0.25, bool bWeightBySize = true) : 
        _eta(eta), _bWeightBySize(bWeightBySize) { /* do nothing */ };
    ~drwnRobustPottsCRFInference() { /* do nothing */ };

 protected:
    void addAuxiliaryVariables(drwnMaxFlow *g, const drwnSegImageInstance* instance) const;

    double addHigherOrderTerms(drwnMaxFlow *g, double higherOrderStrength,
        const drwnSegImageInstance* instance, int alpha, int varOffset = 0) const;

    double computeHigherOrderEnergy(const drwnSegImageInstance *instance,
        double higherOrderStrength) const;
};

// drwnWeightedRobustPottsCRFInference ---------------------------------------

class drwnWeightedRobustPottsCRFInference : public drwnPixelSegCRFInference {
 public:
    double _eta;               

 public:
    drwnWeightedRobustPottsCRFInference(double eta = 0.25) : _eta(eta) { /* do nothing */ };
    ~drwnWeightedRobustPottsCRFInference() { /* do nothing */ };

 protected:
    void addAuxiliaryVariables(drwnMaxFlow *g, const drwnSegImageInstance* instance) const;

    double addHigherOrderTerms(drwnMaxFlow *g, double higherOrderStrength,
        const drwnSegImageInstance* instance, int alpha, int varOffset = 0) const;

    double computeHigherOrderEnergy(const drwnSegImageInstance *instance,
        double higherOrderStrength) const;
};