Facial-Recognition/recognizer.h at master · windsice/Facial-Recognition · GitHub

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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
#ifndef RECOGNIZER_H
#define RECOGNIZER_H

#include <QMainWindow>
#include <QFileDialog>
#include <QSettings>
#include <QDirIterator>
#include <QDebug>
#include <QMessageBox>
#include <QProgressBar>
#include <QThread>
#include <qtconcurrentrun.h>

#include "opencv2/core/core.hpp"
#include "opencv2/contrib/contrib.hpp"
#include "opencv2/highgui/highgui.hpp"

#include "stillobject.h"
#include "detector.h"
#include "XML_creator.h"

#include <iostream>
#include <fstream>
#include <sstream>

using namespace std;
using namespace cv;

namespace Ui {
class Recognizer;
}

class Recognizer : public QMainWindow
{
    Q_OBJECT

signals:
    void TrainingObjects(QString type);

public:
    explicit Recognizer(QWidget *parent = 0);
    ~Recognizer();
    static Mat norm_0_255(InputArray _src);
    void read_csv(const string& filename, vector<Mat>& images, vector<int>& labels, char separator = ';');

    enum TabIndex {TAB_EMPTY,TAB_STILLFACIAL,TAB_LIVEFACIAL,TAB_STILLOBJECT,TAB_XMLCREATOR};

    static const QList<QSize> RESOLUTION;

private slots:
    void onTrainingObjects(QString type);

    //*****************************Preparation*************************
    void on_CVS_Path_Button_clicked();
    void on_CVS_Path_lineEdit_textChanged(const QString &arg1);
    void on_CreateCVS_Button_clicked();

    void on_ClassifierLineEdit_textChanged(const QString &arg1);
    void on_ClassifierPath_toolButton_clicked();
    void on_lineEdit_colorClassifier_textChanged(const QString &arg1);
    void on_toolButton_colorClassifier_clicked();

    void on_LBPH_checkBox_clicked(bool checked);
    void on_LBPH_lineEdit_textChanged(const QString &arg1);
    void on_LBPH_toolButton_clicked();
    void on_EigenFace_checkBox_clicked(bool checked);
    void on_EigenFace_location_toolButton_clicked();
    void on_EigenFace_location_lineEdit_textChanged(const QString &arg1);
    void on_FisherFace_location_toolButton_clicked();
    void on_FisherFace_checkBox_clicked(bool checked);
    void on_FisherFace_location_lineEdit_textChanged(const QString &arg1);

    void on_SaveTrainedModel_pushButton_clicked();
    void on_LoadTrainedModel_pushButton_clicked();
    void on_Train_Button_clicked();
    //******************************************************************

    //*****************************Update*******************************
    void on_UpdateExisting_checkBox_clicked(bool checked);
    void on_UpdateLocation_lineEdit_textChanged(const QString &arg1);
    void on_UpdateLocation_toolButton_clicked();

    void on_TrainNew_pushButton_clicked();
    //******************************************************************

    //*****************************Prediction***************************
    void on_Target_lineEdit_textChanged(const QString &arg1);
    void on_Target_toolButton_clicked();
    void on_Predict_Button_clicked();
    //******************************************************************

    void on_pushButton_liveFacial_clicked();

    void on_pushButton_stillFacial_clicked();

    void on_pushButton_stillObject_clicked();

    void on_pushButton_action_clicked();

    void on_pushButton_setting_clicked();


    void on_pushButton_XMLcreator_clicked();

    void on_toolButton_XMLPositive_clicked();

    void on_toolButton_XMLNegative_clicked();

    void on_lineEdit_XMLObjectName_textChanged(const QString &arg1);

    void on_toolButton_XMLObjectFolder_clicked();

    void on_comboBox_colorClassifierAlgorithm_currentIndexChanged(int index);

    void on_spinBox_colorPercentage_valueChanged(int arg1);

private:
    //change layout based on the operation
    void onOpSel();

    //perform operation based on the settings
    bool passParaToOp();

    void Training();
    void Initialization();
    void TabsInit();
    void SaveSettings();
    void LoadSettings();
    void AlgorithmChecked();

    Ui::Recognizer *settingsUI;
    QProgressBar *statProgress;

    QString CVS_Path;
    QString Output_Folder;
    QString Target_Path;

    QString ClassifierPath;
    QString ColorClassifierPath;
    Ptr<FaceRecognizer> LBPHModel;
    QString LBPHPath;
    Ptr<FaceRecognizer> EigenFaceModel;
    QString EigenFacePath;
    Ptr<FaceRecognizer> FisherFaceModel;
    QString FisherFacePath;

    int img_width;
    int img_height;

    QString UpdateExistingDirectory;

    Mat testSample;
    vector<Mat> images;
    vector<int> labels;
    Detector *detector;
    StillObject *stillObject;
    XML_creator *xml_creator;

    int StackWidgetIndex;
};

#endif // RECOGNIZER_H