整理代码结构

整理代码结构

src/pikaqiu.cpp

@@ -3,25 +3,100 @@
 #include "facenet.h"
 #include <time.h>
 
+void load_emb_csv(int num, vector<vector<mydataFmt>> &vecVec) {
+    for (int i = 0; i < num; ++i) {
+        ifstream inFile("../emb_csv/" + to_string(i) + ".csv", ios::in);
+        string lineStr;
+//                vector<vector<mydataFmt>> strArray;
+        vector<mydataFmt> lineArray;
+        while (getline(inFile, lineStr)) {
+            // 打印整行字符串
+//                    cout << lineStr << endl;
+            // 存成二维表结构
+            stringstream ss(lineStr);
+            string str;
+//                    vector<mydataFmt> lineArray;
+            // 按照逗号分隔
+//                    mydataFmt nnn = 0;
+            while (getline(ss, str, ',')) {
+                lineArray.push_back(atof(str.c_str()));
+//                        cout << str << endl;
+            }
+//                    strArray.push_back(lineArray);
+        }
+        vecVec.push_back(lineArray);
+    }
+}
+
+float compare(vector<mydataFmt> &lineArray0, vector<mydataFmt> &lineArray1) {
+    mydataFmt sum = 0;
+    for (int i = 0; i < Num; ++i) {
+//        cout << lineArray0[i] << "===" << lineArray1[i] << endl;
+        mydataFmt sub = lineArray0[i] - lineArray1[i];
+        mydataFmt square = pow(sub, 2);
+        sum += square;
+    }
+    mydataFmt result = sqrt(sum);
+    return result;
+}
+
+
+void run_mtcnn(Mat &image, vector<Rect> &vecRect) {
+    vector<Point> vecPoint;
+    mtcnn find(image.rows, image.cols);
+    find.findFace(image, vecRect, vecPoint);
+    for (int i = 0; i < vecRect.size(); ++i) {
+        rectangle(image, vecPoint[7 * i + 0], vecPoint[7 * i + 1], Scalar(0, 0, 255), 2, 8, 0);
+        for (int num = 0; num < 5; num++)
+            circle(image, vecPoint[7 * i + num + 2], 2, Scalar(0, 255, 255),
+                   -1);
+    }
+}
+
+void run_facenet(Mat &image, vector<Rect> &vecRect, int csv_num) {
+    for (int i = 0; i < vecRect.size(); ++i) {
+        Mat fourthImage;
+        resize(image(vecRect[i]), fourthImage, Size(299, 299), 0, 0, cv::INTER_LINEAR);
+        facenet ggg;
+//        mydataFmt *o = new mydataFmt[Num];
+        vector<mydataFmt> n;
+        vector<vector<mydataFmt>> o;
+        ggg.run(fourthImage, n, i);
+        load_emb_csv(csv_num, o);
+        for (int j = 0; j < o.size(); ++j) {
+            float result = compare(n, o[j]);
+            cout << result << endl;
+            if (result < 0.85)
+                cout << "it's me" << endl;
+            else
+                cout << "unknow" << endl;
+        }
+    }
+}
+
+
 void run() {
     int b = 0;
     if (b == 0) {
-        Mat image = imread("../40.jpg");
+//        Mat image = imread("../40.jpg");
 //        Mat image = imread("../1.jpeg");
 //        Mat image = imread("../Kong_Weiye.jpg");
-//        Mat image = imread("../Kong_Weiye1.jpg");
+        Mat image = imread("../Kong_Weiye1.jpg");
 //        Mat image = imread("../20.png");
 //        Mat image = imread("../emb_img/0.jpg");
-        mtcnn find(image.rows, image.cols);
+
         clock_t start;
         start = clock();
-        find.findFace(image);
+        vector<Rect> vecRect;
+        run_mtcnn(image, vecRect);
+        run_facenet(image, vecRect, 5);
+
         imshow("result", image);
         imwrite("../result.jpg", image);
         start = clock() - start;
         //    cout<<"time is  "<<start/10e3<<endl;
         cout << "time is " << (double) start / CLOCKS_PER_SEC * 1000 << "ms" << endl;
-        waitKey(0);
+        waitKey(5000);
         image.release();
     } else if (b == 1) {
         Mat image = imread("../10.jpg");
@@ -31,7 +106,7 @@ void run() {
         resize(image, Image, Size(160, 160), 0, 0, cv::INTER_LINEAR);
         facenet ggg;
         mydataFmt *o = new mydataFmt[Num];
-        ggg.run(Image, o, 0);
+//        ggg.run(Image, o, 0);
 //        imshow("result", Image);
         imwrite("../result.jpg", Image);
 
@@ -58,7 +133,9 @@ void run() {
         while (true) {
             start = clock();
             cap >> image;
-            find.findFace(image);
+            vector<Rect> vecRect;
+            vector<Point> vecPoint;
+            find.findFace(image, vecRect, vecPoint);
             imshow("result", image);
             if (waitKey(1) >= 0) break;
             start = clock() - start;