300W資料集測試MTCNN的landmark效果，用提取其中afw資料集337張圖片的預測關鍵點並寫入到txt中，再用測試程式和標註landmark做對比。

程式如下：

1. #include "network.h"
2. #include "mtcnn.h"
3. #include <time.h>
4. #include <fstream>
5. #include <opencv2/opencv.hpp>
6. #pragma comment(lib, "libopenblas.dll.a")
7. using namespace cv;  
8. std::vector<std::string> split(std::string& str, std::string& pattern);  
9. void str2int(int &int_temp, const string &string_temp);  
10. int main()  
11. {  
12.     //因為執行目錄被設定到openblas/x64下了，保證dll能正常載入，這時候圖片路徑就相對要提上去2級  
13.     //Mat im = imread("../../test10.jpg");  
14.     int i = 0;  
15.     string img_dir = "E:/face_alignment/data/300W_test_5points/afw/";  
16.     string name, s="_";  
17.     ifstream infile;  
18.     ofstream outfile;  
19.     infile.open("E:/face_alignment/data/300W_test_5points/afw_mtcnn_test_V2_2.txt"
    );  
20.     outfile.open("E:/face_alignment/data/300W_test_5points/MTCNN_V2_2_test/afw_test.txt");  
21.     while (infile)  
22.     {  
23.         infile >> name;  
24.         vector<string> result = split(name, s);  
25.         cout << i << endl;  
26.         i++;  
27.         //cout << "name: " << result[0] << " s: " << result[1] << endl;  
28.         string shot_name = result[0];  
29.         int decide;  
30.         str2int(decide, result[1]);  
31.         if (decide == 1)  
32.         {  
33.             string image_name = name + ".jpg";  
34.             outfile << shot_name  << endl;  
35.             string image_name_dir = img_dir + image_name;  
36.             cout << image_name_dir << endl;  
37.             Mat im = imread(image_name_dir);  
38.             vector<vector<Point2f>> key_points;  
39.             mtcnn find(im.cols, im.rows);  
40.             vector<Rect> objs = find.detectObject(im, key_points);  
41.             //outfile << objs.size() << endl;  
42.             for (int i = 0; i < objs.size(); ++i)  
43.             {  
44.                 rectangle(im, objs[i], Scalar(0, 255), 2);  
45.                 //outfile << objs[i].x << " " << objs[i].y << " " << objs[i].width << " " << objs[i].height << endl;  
46.             }  
47.             //cout << "num of key_points: " << key_points.size() << endl;  
48.             outfile << key_points.size() << endl;  
49.             for (int i = 0; i < key_points.size(); i++)  
50.             {  
51.                 for (int j = 0; j < key_points[i].size(); j++)  
52.                 {  
53.                     cv::circle(im, key_points[i][j], 1, cv::Scalar(255, 255, 0), 2);  
54.                     outfile << key_points[i][j].x << " " << key_points[i][j].y << " ";  
55.                 }  
56.                 outfile << endl;  
57.             }  
58.             string outdir = "E:/face_alignment/data/300W_test_5points/MTCNN_V2_2_test/afw/";  
59.             string out_image = outdir + shot_name + ".jpg";  
60.             imwrite(out_image, im);  
61.             //imshow("demo", im);  
62.             //waitKey(0);  
63.         }  
64.     }  
65.     infile.close();  
66.     outfile.close();  
67.     return0;  
68. }  
69. //字串分割函式     
70. std::vector<std::string> split(std::string& str, std::string& pattern)  
71. {  
72.     std::string::size_type pos;  
73.     std::vector<std::string> result;  
74.     //str += pattern;//擴充套件字串以方便操作       
75.     int size = str.size();  
76.     pos = str.find(pattern, 0);  
77.     if (pos<size) {  
78.         std::string s1 = str.substr(0, pos);  
79.         std::string s2 = str.substr(pos + 1, size - 1);  
80.         result.push_back(s1);  
81.         result.push_back(s2);  
82.     }  
83.     return result;  
84. }  
85. void str2int(int &int_temp, const string &string_temp)  
86. {  
87.     stringstream stream(string_temp);  
88.     stream >> int_temp;  
89. }  

1. #include <iostream>  
2. #include <stdlib.h>  
3. #include <fstream>  
4. #include <sstream>  
5. #include <string>  
6. #include <vector>  
7. #include <opencv2/opencv.hpp>    
8. using namespace cv;  
9. using namespace std;  
10. std::vector<std::string> split(std::string& str, std::string& pattern);  
11. void str2int(int &int_temp, const string &string_temp);  
12. float computer_error(vector<float> pts_gt, vector<vector<float>> pts_pre);  
13. int main()  
14. {  
15.     int count = 0, pos = 0;  
16.     float acc, thread=0.1;  
17.     string name_list, s = "_";  
18.     string pts_dir = "E:/face_alignment/data/300W_test_5points/afw/";  
19.     ifstream infile_list, infile_pre;  
20.     infile_list.open("E:/face_alignment/data/300W_test_5points/afw_mtcnn_test_V1.txt");  
21.     infile_pre.open("E:/face_alignment/data/300W_test_5points/MTCNN_V1_test/afw_test.txt");  
22.     infile_list >> name_list;  
23.     while (infile_pre)  
24.     {  
25.         string name_pre;  
26.         int num_pre;  
27.         vector<vector<float> > pts_pre;  
28.         infile_pre >> name_pre;  
29.         infile_pre >> num_pre;  
30.         pts_pre.resize(num_pre);  
31.         for (int i = 0; i < num_pre; i++)  
32.         {  
33.             pts_pre[i].resize(10);  
34.         }  
35.         for (int j = 0; j < num_pre; j++)  
36.         {  
37.             infile_pre >> pts_pre[j][0] >> pts_pre[j][1] >> pts_pre[j][2] >> pts_pre[j][3] >> pts_pre[j][4] >> pts_pre[j][5] >> pts_pre[j][6] >> pts_pre[j][7] >> pts_pre[j][8] >> pts_pre[j][9];  
38.         }  
39.         //for (int i = 0; i < num_pre; i++)  
40.         //{  
41.         //  for (int j = 0; j < 10; j++)  
42.         //  {  
43.         //      cout << pts_pre[i][j] << " ";  
44.         //  }  
45.         //  cout << endl;  
46.         //}  
47.         // read gt file  
48.         while (infile_list)  
49.         {  
50.             vector<string> result = split(name_list, s);  
51.             string name_gt = result[0];  
52.             if (name_gt.compare(name_pre) == 0)  
53.             {  
54.                 count++;  
55.                 cout << count << endl;  
56.                 vector<float> pts_gt;  
57.                 pts_gt.resize(10);  
58.                 string pts_dir_name = pts_dir + name_list + ".pts";  
59.                 ifstream infile_pts;  
60.                 infile_pts.open(pts_dir_name);  
61.                 string ss;  
62.                 int yy;  
63.                 infile_pts >> ss >> yy;  
64.                 infile_pts >> ss >> yy;  
65.                 infile_pts >> ss;  
66.                 for (int i = 0; i < 5; i++)  
67.                 {  
68.                     infile_pts >> pts_gt[i*2] >> pts_gt[i*2+1];  
69.                 }  
70.                 infile_pts.close();  
71.                 float error = computer_error(pts_gt, pts_pre);  
72.                 error = error / 5.0;  
73.                 if (error <= thread)  
74.                     pos++;  
75.                 cout << error << " " << endl;  
76.                 infile_list >> name_list;  
77.                 //cout << name_list << endl;  
78.             }  
79.             else
80.                 break;  
81.         }     
82.     }  
83.     acc = float(pos) / float(count);  
84.     cout << "accury: " << acc << endl;  
85.     infile_list.close();  
86.     infile_pre.close();  
87. }  
88. // computer alinment loss  
89. float computer_error(vector<float> pts_gt, vector<vector<float>> pts_pre)  
90. {  
91.     if (pts_pre.size() == 0)  
92.         return10;  
93.     float RMSE, d_outer, align_loss;  
94.     d_outer = sqrt((pts_gt[0] - pts_gt[8])*(pts_gt[0] - pts_gt[8]) + (pts_gt[1] - pts_gt[9])*(pts_gt[1] - pts_gt[9]));  
95.     for (int i = 0; i < pts_pre.size(); i++)  
96.     {  
97.         RMSE = 0;  
98.         for (int j = 0; j < 5; j++)  
99.         {  
100.             RMSE += sqrt((pts_gt[2 * j] - pts_pre[i][2 * j])*(pts_gt[2 * j] - pts_pre[i][2 * j]) + (pts_gt[2 * j + 1] - pts_pre[i][2 * j + 1])*(pts_gt[2 * j + 1] - pts_pre[i][2 * j + 1]));  
101.         }  
102.         RMSE = RMSE / d_outer;  
103.         if (i == 0)  
104.         {  
105.             align_loss = RMSE;  
106.         }  
107.         else
108.         {  
109.             if (align_loss > RMSE)  
110.                 align_loss = RMSE;  
111.         }  
112.     }  
113.     return align_loss;  
114. }  
115. //字串分割函式     
116. std::vector<std::string> split(std::string& str, std::string& pattern)  
117. {  
118.     std::string::size_type pos;  
119.     std::vector<std::string> result;  
120.     //str += pattern;//擴充套件字串以方便操作       
121.     int size = str.size();  
122.     pos = str.find(pattern, 0);  
123.     if (pos<size) {  
124.         std::string s1 = str.substr(0, pos);  
125.         std::string s2 = str.substr(pos + 1, size - 1);  
126.         result.push_back(s1);  
127.         result.push_back(s2);  
128.     }  
129.     return result;  
130. }  
131. void str2int(int &int_temp, const string &string_temp)  
132. {  
133.     stringstream stream(string_temp);  
134.     stream >> int_temp;  
135. }