From c34476f4d9a513e2dc5aecdfa40fe8aa459ffcea Mon Sep 17 00:00:00 2001
From: ChrisKong <609027949@qq.com>
Date: Wed, 25 Dec 2019 17:18:34 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E3=80=80=E4=BF=9D=E5=AD=98em?=
 =?UTF-8?q?b=E5=80=BC=E3=80=80=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

添加　保存emb值　功能
---
 .gitignore      |  2 ++
 src/pikaqiu.cpp | 27 ++++++++++++++++++++++-----
 2 files changed, 24 insertions(+), 5 deletions(-)

diff --git a/.gitignore b/.gitignore
index 5e0c20e..4aec7e0 100755
--- a/.gitignore
+++ b/.gitignore
@@ -13,3 +13,5 @@ model_128.zip
 model_512.zip
 emb_csv/*.csv
 emb_img/*.jpg
+
+kkk.jpg
diff --git a/src/pikaqiu.cpp b/src/pikaqiu.cpp
index 7df0e51..3a8d879 100755
--- a/src/pikaqiu.cpp
+++ b/src/pikaqiu.cpp
@@ -28,6 +28,22 @@ void load_emb_csv(int num, vector<vector<mydataFmt>> &vecVec) {
     }
 }
 
+void write_emb_csv(vector<mydataFmt> &o, int num) {
+    ofstream outFile;
+    outFile.open("../emb_csv/" + to_string(num + 1) + ".csv", ios::out); // 打开模式可省略
+    for (int l = 0; l < Num; ++l) {
+//            cout << o[l] << endl;
+        if (l == Num - 1) {
+            outFile << o[l];
+        } else {
+            outFile << o[l] << ',';
+        }
+    }
+    outFile << endl;
+    outFile.close();
+    cout << "write over!" << endl;
+}
+
 float compare(vector<mydataFmt> &lineArray0, vector<mydataFmt> &lineArray1) {
     mydataFmt sum = 0;
     for (int i = 0; i < Num; ++i) {
@@ -40,7 +56,6 @@ float compare(vector<mydataFmt> &lineArray0, vector<mydataFmt> &lineArray1) {
     return result;
 }
 
-
 void run_mtcnn(Mat &image, vector<Rect> &vecRect) {
     vector<Point> vecPoint;
     mtcnn find(image.rows, image.cols);
@@ -62,6 +77,9 @@ void run_facenet(Mat &image, vector<Rect> &vecRect, int csv_num) {
         vector<mydataFmt> n;
         vector<vector<mydataFmt>> o;
         ggg.run(fourthImage, n, i);
+
+//        write_emb_csv(n, i);
+//        return;
         load_emb_csv(csv_num, o);
         for (int j = 0; j < o.size(); ++j) {
             float result = compare(n, o[j]);
@@ -74,14 +92,13 @@ void run_facenet(Mat &image, vector<Rect> &vecRect, int csv_num) {
     }
 }
 
-
 void run() {
     int b = 0;
     if (b == 0) {
 //        Mat image = imread("../40.jpg");
-//        Mat image = imread("../1.jpeg");
+//        Mat image = imread("../3.jpeg");
 //        Mat image = imread("../Kong_Weiye.jpg");
-        Mat image = imread("../Kong_Weiye1.jpg");
+        Mat image = imread("../kkk.jpg");
 //        Mat image = imread("../20.png");
 //        Mat image = imread("../emb_img/0.jpg");
 
@@ -89,7 +106,7 @@ void run() {
         start = clock();
         vector<Rect> vecRect;
         run_mtcnn(image, vecRect);
-        run_facenet(image, vecRect, 5);
+        run_facenet(image, vecRect, 13);
 
         imshow("result", image);
         imwrite("../result.jpg", image);