Newer
Older
#include <iostream>
#include <string>
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui.hpp"
#include "opencv2/core/core.hpp"
using namespace cv;
Mat picture, picture_grayscale;
char key;
int picture_height;
int picture_width;
int main() {
//Starting the video
VideoCapture videoCapture(0);
if (!videoCapture.isOpened()) {
cout << "Unable to open video file." << endl;
return 1;
}
namedWindow("Webcam", CV_WINDOW_AUTOSIZE);
while(true) {
Mat frame;
videoCapture.retrieve(frame);
bool success = videoCapture.read(frame);
if (!success) {
cout << "Could not read from video file" << endl;
return 1;
}
//show some test text
Mat frame_with_text = frame.clone();
putText(frame_with_text, "Press Spacebar to take a picture", Point2f(110,100), FONT_HERSHEY_SIMPLEX, 2.0, Scalar(255,0,0,0), 3);
putText(frame_with_text, "Press ESC to close the application", Point2f(90,600), FONT_HERSHEY_SIMPLEX, 2.0, Scalar(0,0,255,255), 3);
imshow("Webcam", frame_with_text);
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
key = waitKey(30);
if (key == 27) { //escape key pressed: stop program
cout << "ESC pressed. Program closing..." << endl;
break;
}
else if (key == ' ') { //spacebar pressed: take a picture
picture = frame;
key = -1;
picture_width = frame.rows;
picture_height = frame.cols;
//Convert to grayscale
picture_grayscale = picture.clone();
cvtColor(picture, picture_grayscale, CV_BGR2GRAY);
// generate eigenface
vector<Mat> images;
vector<int> labels;
labels.push_back(0);
images.push_back(picture_grayscale);
Ptr<FaceRecognizer> model = createEigenFaceRecognizer();
model->train(images, labels);
cout << "Face Recognizer created" << endl;
int predictedLabel = model->predict(picture_grayscale);
cout << "Predicted label: " << predictedLabel << endl;
while (true) {
Mat picture_with_text = picture.clone();
putText(picture_with_text, "Press 's' to save", Point2f(375,100), FONT_HERSHEY_SIMPLEX, 2.0, Scalar(255,0,0,0), 3);
putText(picture_with_text, "Press ESC/Spacebar to return", Point2f(160,600), FONT_HERSHEY_SIMPLEX, 2.0, Scalar(0,0,255), 3);
imshow("Webcam", picture_with_text);
key = waitKey(30);
if (key == 27 || key == 32) { //spacebar or ESC pressed
cout << "ESC or SPACE pressed. Returning to video..." << endl;
break;
} else if (key == 115) {
string name;
string directory = "./images/";
cout << "Please select a name for the image." << endl;
getline(cin, name);
directory.append(name);
directory.append(".png");
Point org;
org.x = 15;
org.y = 100;
putText(picture_with_text, name, org, FONT_HERSHEY_SIMPLEX, 2, Scalar(0, 0, 255), 3);
bool maybe = imwrite(directory, picture);
cout << "s was pressed. saving image, success: " << maybe << endl;
}
}
}
}
return 0;