FacialRec.cpp

//This is our project

#include <iostream>
#include <string>

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

using namespace std;
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);

		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;

			//Get stored images
			vector<Mat> images;
			vector<int> labels;

			Mat load_image, load_image_grayscale;
			string filename = "";
			string directory = "./myfaces/";
			string subdirectory = "";
			for (int j = 0; j < 2; j++) {
				subdirectory = to_string(j) + "/";
				int i = 0;
				while (true) {
					filename = to_string(i);
					string path = directory + subdirectory + filename + ".png";
					cout << "path: " << path << endl;
					load_image = imread(path, 1);

					if(!load_image.data)
					{
						printf( " No image data \n " );
						break;
					}

					load_image_grayscale = load_image.clone();
					//TODO: The following line isn't working for some reason
					cvtColor(load_image, load_image_grayscale, CV_BGR2GRAY);
					images.push_back(load_image_grayscale);
					labels.push_back(0);
					i++;
				}
			}
			//Convert to grayscale
			picture_grayscale = picture.clone();
			cvtColor(picture, picture_grayscale, CV_BGR2GRAY);

			// generate eigenface
			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 = "./myfaces/";
		          	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;
}