-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathMain.cpp
122 lines (116 loc) · 4.01 KB
/
Main.cpp
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
#include "Haar.h"
#include "Config.h"
#include "lbp.h"
#ifdef _DEBUG
#pragma comment(lib, "opencv_core249d.lib")
#pragma comment(lib, "opencv_calib3d249d.lib")
#pragma comment(lib, "opencv_contrib249d.lib")
#pragma comment(lib, "opencv_features2d249d.lib")
#pragma comment(lib, "opencv_flann249d.lib")
#pragma comment(lib, "opencv_gpu249d.lib")
#pragma comment(lib, "opencv_highgui249d.lib")
#pragma comment(lib, "opencv_imgproc249d.lib")
#pragma comment(lib, "opencv_legacy249d.lib")
#pragma comment(lib, "opencv_ml249d.lib")
#pragma comment(lib, "opencv_objdetect249d.lib")
#pragma comment(lib, "opencv_ts249d.lib")
#pragma comment(lib, "opencv_video249d.lib")
#pragma comment(lib, "gurobi_c++mdd2012.lib")
#pragma comment(lib, "gurobi56.lib")
#else
#pragma comment(lib, "opencv_calib3d249.lib")
#pragma comment(lib, "opencv_contrib249.lib")
#pragma comment(lib, "opencv_core249.lib")
#pragma comment(lib, "opencv_features2d249.lib")
#pragma comment(lib, "opencv_flann249.lib")
#pragma comment(lib, "opencv_gpu249.lib")
#pragma comment(lib, "opencv_highgui249.lib")
#pragma comment(lib, "opencv_imgproc249.lib")
#pragma comment(lib, "opencv_legacy249.lib")
#pragma comment(lib, "opencv_ml249.lib")
#pragma comment(lib, "opencv_objdetect249.lib")
#pragma comment(lib, "opencv_ts249.lib")
#pragma comment(lib, "opencv_video249.lib")
#pragma comment(lib, "gurobi_c++md2012.lib")
#pragma comment(lib, "gurobi56.lib")
#endif
int main()
{
// read config file
string configPath = "config.txt";
Config conf(configPath);
cout << conf << endl;
VideoCapture cap;
vector<Haar> feat;
feat.resize(100);
// 在LBP特征的基础上随机生成100种haar特征
for (int i=0; i<100; i++)
{
// LBP特征中每个图像块被归一化为32*32，因此这里的patchwidth和patchheight都是32
feat[i] = HaarGenerate(patchwidth,patchheight);
}
int startFrame = -1;
int endFrame = -1;
string imgFormat;
string framesFilePath = conf.sequenceBasePath+"\\"+conf.sequenceName+"\\"+conf.sequenceName+"_frames.txt";
ifstream framesFile(framesFilePath.c_str(), ios::in);
string framesLine;
getline(framesFile, framesLine);
sscanf(framesLine.c_str(), "%d,%d", &startFrame, &endFrame);
imgFormat = conf.sequenceBasePath+"\\"+conf.sequenceName+"\\imgs\\%04d.jpg";
string gtFilePath = conf.sequenceBasePath+"\\"+conf.sequenceName+"\\"+conf.sequenceName+"_gt.txt";
ifstream gtFile(gtFilePath.c_str(), ios::in);
string gtLine;
getline(gtFile, gtLine);
float xmin = -1.f;
float ymin = -1.f;
float width = -1.f;
float height = -1.f;
sscanf(gtLine.c_str(), "%f,%f,%f,%f", &xmin, &ymin, &width, &height);
// initRect初始检测框。
Rect initRect(xmin,ymin,width,height);
oriRect.x = initRect.x;
oriRect.y = initRect.y;
oriRect.width = initRect.width;
oriRect.height = initRect.height;
// rect1表示检测框
Rect rect1(initRect);
bool paused = false;
bool doInitialise = false;
vector<Rect> res;
// res用来记录历史检测框
res.push_back(rect1);
for (int frameInd = startFrame; frameInd <= endFrame-1; ++frameInd)
{
long t1 = GetTickCount();
//Mat frame;
char imgPath[256];
sprintf(imgPath, imgFormat.c_str(), frameInd);
cout<<imgPath;
string path(imgPath);
// frameOrig表示当前帧
Mat frameOrig = imread(path, 0);
vectorf w = LearnApp(frameOrig, rect1, feat);
sprintf(imgPath, imgFormat.c_str(), frameInd+1);
string path2(imgPath);
frameOrig = imread(path2, 0);
rect1 = Locate(frameOrig,rect1,feat,w);
namedWindow("im",1);
cvtColor(frameOrig,frameOrig,CV_GRAY2BGR);
putText(frameOrig,int2str(frameInd,4),Point(20,20),1,2,Scalar(0,255,0));
rectangle(frameOrig,rect1,Scalar(0,255,0),2);
imshow("im",frameOrig);
waitKey(1);
res.push_back(rect1);
long t2 = GetTickCount();
cout<<t2-t1<<endl;
}
string name = conf.sequenceName + int2str(conf.seed,2) + int2str(6,1) +".txt";
ofstream outFile(name.c_str());
for (int i=0; i<res.size(); i++)
{
outFile<<res[i].x<<' '<<res[i].y<<' '<<res[i].width<<' '<<res[i].height<<endl;
}
system("pause");
return EXIT_SUCCESS;
}