/media/Data/chenxing/myProj/asmlib-opencv/src/shapemodel.h

Go to the documentation of this file.

#ifndef SHAPEMODEL_H
#define SHAPEMODEL_H

#include "modelimage.h"
#include "cv.h"
#include "modelfile.h"
#include <vector>
using std::vector;
using cv::PCA;

namespace StatModel {

struct FitResult{
    Mat_< double > params;

    SimilarityTrans transformation;
};


class ShapeModel
{
public:
    ShapeModel();

    virtual void saveToFile(ModelFile &file);

    virtual void loadFromFile(ModelFile &file);

    void readTrainData(const char *listFileName);

    void loadShapeInfo(const char *shapeFileName);


    void setPyramidLevel(int l){ pyramidLevel = l; }


    void projectParamToShape(const Mat_<double> & paramVec, ShapeVec &shapeVec);


    void projectShapeToParam(const ShapeVec & shapeVec, Mat_<double> &paramVec);

    ShapeInfo & getShapeInfo(){ return shapeInfo;}

    Mat_< double > normalizeParam(const Mat_<double> &p){
        Mat_<double> n = p.clone();
        for (int i=0; i<p.rows; i++)
            n(i, 0)/= 3*sqrt(pcaShape->eigenvalues.at<double>(i, 0));
        return n;
    }

    Mat_< double > reConFromNorm(const Mat_<double> &p){
        Mat_<double> n = p.clone();
        for (int i=0; i<p.rows; i++)
            n(i, 0)*= 3*sqrt(pcaShape->eigenvalues.at<double>(i, 0));
        return n;
    }


    // For viewing the model
    struct ModelViewInfo
    {
        vector< int > vList;
        int curParam;
        void *pModel;
    };

    void viewShapeModel();

    void viewShapeModelUpdate(ModelViewInfo *pInfo);
protected:
    int pyramidLevel;

    int nMarkPoints;

    int nTrain;

    vector<ModelImage> imageSet;

    ShapeInfo shapeInfo;

    ShapeVec meanShape;

    void clampParamVec( Mat_< double > &paramVec );

    PCA *pcaShape;
    int nShapeParams;

    double sigma2;

    PCA *pcaFullShape;


    void buildModel(const string& shapeDefFile, const string& ptsListFile);
private:
    void buildPCA();

    void alignShapes();

    void preparePatterns();

protected:
    double searchYOffset;

    double searchXOffset;

    double searchWScale;
    double searchHScale;

    double searchStepAreaRatio;

    double searchScaleRatio;
    double searchInitXOffset;
    double searchInitYOffset;
};
} // Namespace
#endif // SHAPEMODEL_H