-
Notifications
You must be signed in to change notification settings - Fork 231
/
Copy pathRandomForest.h
executable file
·50 lines (41 loc) · 1.34 KB
/
RandomForest.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
//
// 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_;
double 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(const std::vector<cv::Mat_<uchar> >& images,
const std::vector<cv::Mat_<double> >& ground_truth_shapes,
const std::vector<cv::Mat_<double> >& current_shapes,
const std::vector<BoundingBox> & bounding_boxs,
const cv::Mat_<double>& mean_shape,
const std::vector<cv::Mat_<double> >& shapes_residual,
int stages
);
void Read(std::ifstream& fin);
void Write(std::ofstream& fout);
};
#endif /* defined(__myopencv__RandomForest__) */