basic neural network in C

Framework/data_loading.cpp

#include "header_file.h" int ReverseInt(int i) { unsigned char ch1, ch2, ch3, ch4; ch1 = i & 255; ch2 = (i >> 8) & 255; ch3 = (i >> 16) & 255; ch4 = (i >> 24) & 255; return((int)ch1 << 24) + ((int)ch2 << 16) + ((int)ch3 << 8) + ch4; } //Load MNIST images function void Loading_Mnist_images(string filename, vector<cv::Mat> &images) { ifstream file(filename, ios::in | ios::binary); if (file.is_open() == false) { cout << "open mnist image file error!" << endl; return; } else if (file.is_open() == true) { int magic_number = 0; int number_of_images = 0; int n_rows = 0; int n_cols = 0; //load sizeof(magic_number) to magic_number file.read((char*)&magic_number, sizeof(magic_number)); magic_number = ReverseInt(magic_number); //load number_of_images file.read((char*)&number_of_images, sizeof(number_of_images)); number_of_images = ReverseInt(number_of_images); //Find image Height file.read((char*)&n_rows, sizeof(n_rows)); n_rows = ReverseInt(n_rows); //Find image Width file.read((char*)&n_cols, sizeof(n_cols)); n_cols = ReverseInt(n_cols); cout << "Total number of images are: " << number_of_images << endl; // read image and save to images for (int i = 0; i < number_of_images; ++i) { //cv::Mat current_image = cv::Mat::zeros(n_rows, n_cols, CV_8UC1); cv::Mat current_image = cv::Mat::zeros(1,n_rows* n_cols, CV_8UC1); int current_index = 0; for (int r = 0; r < n_rows; ++r) { for (int c = 0; c < n_cols; ++c) { unsigned char temp = 0; file.read((char*)&temp, sizeof(temp)); /*current_image.at<uchar>(r, c) = (int)temp;*/ current_image.at<uchar>(current_index) = (int)temp; current_index++; } } images.push_back(current_image); } } } // Load Mnist label function void Load_Mnist_Label(string filename, vector<int> &labels) { ifstream file(filename, ios::in | ios::binary); if (file.is_open()) { int magic_number = 0; int number_of_labels = 0; file.read((char*)&magic_number, sizeof(magic_number)); magic_number = ReverseInt(magic_number); file.read((char*)&number_of_labels, sizeof(number_of_labels)); number_of_labels = ReverseInt(number_of_labels); cout << "Total number of labels are: " << number_of_labels << endl; for (int i = 0; i < number_of_labels; ++i) { unsigned char current_label = 0; file.read((char*)&current_label, sizeof(current_label)); labels.push_back(current_label); } } } // Give a name for each images function string GetImageName(int number, int arr[]) { string str1, str2; for (int i = 0; i < 10; i++) { if (number == i) { arr[i]++; char ch1[10]; sprintf(ch1, "%d", arr[i]); str1 = std::string(ch1); if (arr[i] < 10) { str1 = "0000" + str1; } else if (arr[i] < 100) { str1 = "000" + str1; } else if (arr[i] < 1000) { str1 = "00" + str1; } else if (arr[i] < 10000) { str1 = "0" + str1; } break; } } char ch2[10]; sprintf(ch2, "%d", number); str2 = std::string(ch2); str2 = str2 + "_" + str1; return str2; } // writing training images to train image folder void WritingTrainingIimage(vector<cv::Mat> &train_images_vector, vector<int> &train_labels_vector) { /******************************* writing training images to train image folder **************************************/ int train_count_digits[10]; for (int i = 0; i < 10; i++) train_count_digits[i] = 0; string save_train_images_path = "train_images/"; //path for train images for (int i = 0; i < train_images_vector.size(); i++) { int number = train_labels_vector[i]; string image_name = GetImageName(number, train_count_digits); image_name = save_train_images_path + image_name + ".jpg"; cv::imwrite(image_name, train_images_vector[i].reshape(0,28)); if (i % 5000 == 0) { cout << "Saving " << i << " images" << endl; } if (i == 10) { cout << "10 images are written into train images folder. If you want to continuou to write image, click Y then click Enter. Else click N then click Enter. " << std::endl; char training_button; cin >> training_button; if (training_button == 'y') { cout << "Continue to write image" << endl; continue; } else { cout << "Stop to write image" << endl; break; } } } } // writing testing images to test image folder void WritingTestingIimage(vector<cv::Mat> &test_images_vector, vector<int> &test_labels_vector) { /******************************* writing testinging image to test image folder **************************************/ int test_count_digits[10]; for (int i = 0; i < 10; i++) test_count_digits[i] = 0; string save_test_images_path = "test_images/"; //path for test images for (int i = 0; i < test_images_vector.size(); i++) { int number = test_labels_vector[i]; string image_name = GetImageName(number, test_count_digits); image_name = save_test_images_path + image_name + ".jpg"; cv::imwrite(image_name, test_images_vector[i].reshape(0, 28)); if (i % 5000 == 0) { cout << "Saving " << i << " images" << endl; } if (i == 10) { cout << "10 images are written into test images folder. If you want to continuou to write image, click Y then click Enter. Else click N then click Enter. " << std::endl; char testing_button; cin >> testing_button; if (testing_button == 'y') { cout << "Continue to write image" << endl; continue; } else { cout << "Stop to write image" << endl; break; } } } }