98 lines
2.3 KiB
C++
98 lines
2.3 KiB
C++
#ifndef __MAXPRODBP_H__
|
|
#define __MAXPRODBP_H__
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <assert.h>
|
|
#include "mrf.h"
|
|
#include "LinkedBlockList.h"
|
|
#include "regions-new.h"
|
|
|
|
#define FloatType float
|
|
#define FLOATTYPE float
|
|
class MaxProdBP;
|
|
|
|
class MaxProdBP : public MRF{
|
|
public:
|
|
MaxProdBP(int width, int height, int nLabels, EnergyFunction *eng);
|
|
MaxProdBP(int nPixels, int nLabels,EnergyFunction *eng);
|
|
~MaxProdBP();
|
|
void setNeighbors(int pix1, int pix2, CostVal weight);
|
|
Label getLabel(int pixel){return(m_answer[pixel]);};
|
|
void setLabel(int pixel,Label label){m_answer[pixel] = label;};
|
|
Label* getAnswerPtr(){return(m_answer);};
|
|
void clearAnswer();
|
|
void setParameters(int , void *){printf("No optional parameters to set");}
|
|
EnergyVal smoothnessEnergy();
|
|
EnergyVal dataEnergy();
|
|
EnergyFunction *getEnergyFunction();
|
|
int getWidth();
|
|
int getHeight();
|
|
FLOATTYPE *getScratchMatrix();
|
|
int getNLabels();
|
|
bool varWeights();
|
|
void setExpScale(int expScale);
|
|
friend void getPsiMat(OneNodeCluster &cluster, FLOATTYPE *&destMatrix,
|
|
int r, int c, MaxProdBP *mrf, int direction, FLOATTYPE &var_weight);
|
|
|
|
InputType getSmoothType();
|
|
FLOATTYPE getExpV(int i);
|
|
FLOATTYPE *getExpV();
|
|
|
|
CostVal getHorizWeight(int r, int c);
|
|
CostVal getVertWeight(int r, int c);
|
|
|
|
|
|
CostVal m_lambda;
|
|
CostVal m_smoothMax;
|
|
int m_smoothExp;
|
|
|
|
enum //Borrowed from BP-S.h by vnk
|
|
{
|
|
NONE,
|
|
L1,
|
|
L2,
|
|
FIXED_MATRIX,
|
|
GENERAL,
|
|
BINARY,
|
|
} m_type;
|
|
|
|
protected:
|
|
void setData(DataCostFn dcost);
|
|
void setData(CostVal* data);
|
|
void setSmoothness(SmoothCostGeneralFn cost);
|
|
void setSmoothness(CostVal* V);
|
|
void setSmoothness(int smoothExp,CostVal smoothMax, CostVal lambda);
|
|
void setCues(CostVal* hCue, CostVal* vCue);
|
|
void initializeAlg();
|
|
void BPinitializeAlg();
|
|
void optimizeAlg(int nIterations);
|
|
|
|
private:
|
|
Label *m_answer;
|
|
CostVal *m_V;
|
|
CostVal *m_D;
|
|
CostVal *m_horizWeights;
|
|
CostVal *m_vertWeights;
|
|
FLOATTYPE m_exp_scale;
|
|
DataCostFn m_dataFn;
|
|
SmoothCostGeneralFn m_smoothFn;
|
|
bool m_needToFreeV;
|
|
FLOATTYPE *m_scratchMatrix;
|
|
FLOATTYPE *m_ExpData;
|
|
FLOATTYPE *m_message_chunk;
|
|
OneNodeCluster *nodeArray;
|
|
typedef struct NeighborStruct {
|
|
int to_node;
|
|
CostVal weight;
|
|
} Neighbor;
|
|
|
|
LinkedBlockList *m_neighbors;
|
|
};
|
|
|
|
|
|
#endif /* __ICM_H__ */
|
|
|
|
|