Hard-coded to read the first two subdirectories of a myfaces directory and add...

Hard-coded to read the first two subdirectories of a myfaces directory and add any pictures to the FaceRecognizer

FacialRec.cpp

@@ -58,13 +58,42 @@ int main() {
 			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
-			vector<Mat> images;
-			vector<int> labels;
 			labels.push_back(0);
 			images.push_back(picture_grayscale);
 			
@@ -88,7 +117,7 @@ int main() {
 					break;
 				} else if (key == 115) {
 		        	string name;
-		        	string directory = "./images/";
+		        	string directory = "./myfaces/";
 		          	cout << "Please select a name for the image." << endl;
 		          	getline(cin, name);
 		          	directory.append(name);