randomforest.h

//
//  RandomForest.h
//  myopencv
//
//  Created by lequan on 1/24/15.
//  Copyright (c) 2015 lequan. All rights reserved.
//

#ifndef __myopencv__RandomForest__
#define __myopencv__RandomForest__

#include "Tree.h"

class RandomForest{
public:
    std::vector<std::vector<Tree> > rfs_;
    int max_numtrees_;
    int num_landmark_;
    int max_depth_;
    int stages_;
    float overlap_ratio_;
    
    
    RandomForest(){
        max_numtrees_ = global_params.max_numtrees;
        num_landmark_ = global_params.landmark_num;
        max_depth_    = global_params.max_depth;
        overlap_ratio_ = global_params.bagging_overlap;
        
        // resize the trees
        rfs_.resize(num_landmark_);
        for (int i=0;i<num_landmark_;i++){
            rfs_[i].resize(max_numtrees_);
        }
    }
    void Train(std::vector<cv::Mat_<uchar> >& images,
			   std::vector<int>& augmented_images,
               std::vector<cv::Mat_<float> >& ground_truth_shapes,
			   std::vector<int>& ground_truth_faces,
               std::vector<cv::Mat_<float> >& current_shapes,
			   std::vector<float>& current_fi,
			   std::vector<double>& current_weight,
               std::vector<BoundingBox> & bounding_boxs,
               const cv::Mat_<float>& mean_shape,
               std::vector<cv::Mat_<float> >& shapes_residual,
               int stages,
			   std::vector<RandomForest>& RandomForest_,
			   std::vector<std::vector<struct model*> > Models_,
			   int posLenth
               );
    void Read(std::ifstream& fin);
    void Write(std::ofstream& fout);
};


#endif /* defined(__myopencv__RandomForest__) */