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

Go to the documentation of this file.

#ifndef ASMMODEL_H
#define ASMMODEL_H
#include "shapemodel.h"
#include "cv.h"


namespace StatModel {

using cv::Point_;
using cv::Mat_;

#define ASM_FIT_VERBOSE_NOVERBOSE 0
#define ASM_FIT_VERBOSE_AT_LEVEL 1
#define ASM_FIT_VERBOSE_AT_ITERATION 2
#define ASM_FIT_VERBOSE_AT_POINT 3

class ASMModel;



class ASMFitResult : public FitResult {
private:
    ASMModel *asmModel;
public:
    ASMFitResult(ASMModel *model): asmModel(model) {}

    void toPointList(vector< Point_<int> > &pV) const;
};


class ASMModel : public ShapeModel
{
private:
    vector< vector< Mat_< double > > > iCovarG;

    vector< vector< Mat_< double > > > meanG;

    int localFeatureRad;

    int ns;

    void buildLocalDiffStructure();

    PCA pcaPyr[3];
    double sigma2Pyr[3];

public:
    ASMModel():localFeatureRad(4), ns(6){}

    ASMModel(const string& filename) { loadFromFile(filename); }


    vector< ASMFitResult > fitAll(
        const Mat & img,
        const vector< cv::Rect > & detectedObjs,
        int verbose=0);


    ASMFitResult fit(const Mat & img, int verbose=0);

    void buildModel(const string& shapeDefFile, const string& ptsListFile);

    void saveToFile(const string& filename);

    void loadFromFile(const string& filename);

    void showResult(Mat &img, const vector< ASMFitResult > &res);

private:

    void findParamForShape(const ShapeVec &Y, ASMFitResult & fitResult);


    void findParamForShapeBTSM(const ShapeVec &Y, const ShapeVec &Y_old,
                               ASMFitResult & fitResult, ASMFitResult &b_old, int l);

    void saveToFile(ModelFile &file);

    void loadFromFile(ModelFile &file);
};

} // Namespace
#endif // ASMMODEL_H