MNIST数据集的生成

#include "Image_to_BinaryData.h" #include <atlfile.h> #include <string.h> #include <cstdlib> #include <iostream> #include <fstream> using namespace std; using namespace cv; void Image2BinaryData::ReverseInt(ofstream &file, int i) { char ch1 = (char)(i >> 24); char ch2 = (char)((i << 8) >> 24); char ch3 = (char)((i << 16) >> 24); char ch4 = (char)((i << 24) >> 24); file.write((char*)&ch1, sizeof(ch4)); file.write((char*)&ch2, sizeof(ch3)); file.write((char*)&ch3, sizeof(ch2)); file.write((char*)&ch4, sizeof(ch1)); } vector<string> Image2BinaryData::getFileLists(string file_folder) { /* param file_folder: 文件夹路径 */ file_folder += "/*.*"; const char *mystr = file_folder.c_str(); vector<string> flist; string lineStr; vector<string> extendName; //设置文件扩展名 extendName.push_back("JPG"); extendName.push_back("jpg"); extendName.push_back("bmp"); extendName.push_back("png"); extendName.push_back("tiff"); HANDLE file; WIN32_FIND_DATA fileData; char line[1024]; wchar_t fn[1000]; mbstowcs(fn, mystr, 999); //LPCSTR fn1 = (LPCSTR)fn; file = FindFirstFile(fn, &fileData); FindNextFile(file, &fileData); while (FindNextFile(file, &fileData)) { wcstombs(line, (const wchar_t*)fileData.cFileName, 259); lineStr = line; //remove the files which are not images for (int i = 0; i < 4; i++) { if (lineStr.find(extendName[i]) < 999) { flist.push_back(lineStr); break; } } } return flist; } /* 从文件的绝对路径当中得到文件名 例如："C:\\Users\\lyf\\Desktop\\test.txt"，返回"test.txt" */ string Image2BinaryData::getFileName(string & filename) { /* param filename: 文件的绝对路径 */ int iBeginIndex = filename.find_last_of("\\") + 1; int iEndIndex = filename.length(); return filename.substr(iBeginIndex, iEndIndex - iBeginIndex); cout << "Done!" << endl; } void Image2BinaryData::ReadImage(string filefolder, vector<string>& img_Lists, vector<int>& img_Labels, vector<cv::Mat> &ImagesMat) { const int size_list = img_Lists.size(); for (int i = 0; i < size_list; ++i) { string curPath = filefolder + "\\" + img_Lists[i]; Mat image = imread(curPath, IMREAD_UNCHANGED); ImagesMat.push_back(image); char label = img_Lists[i][0]; img_Labels[i] = label - '0'; printf("正在读取图片，请稍等: %.2lf%%\r", i*100.0 / (size_list - 1)); } printf("\n图片读取完毕!\n\n"); } void Image2BinaryData::Image2BinaryFile(string filefolder, vector<cv::Mat>& ImagesMat, vector<int>& img_Labels) { ofstream file(filefolder, ios::binary); int idx = filefolder.find_last_of("\\") + 1; string subName = filefolder.substr(idx); if (file.is_open()) { cout << subName<< "文件创建成功." << endl; int magic_number = 2051; int number_of_images = img_Labels.size(); int n_rows = Height; int n_cols = Width; ReverseInt(file, magic_number); ReverseInt(file, number_of_images); ReverseInt(file, n_rows); ReverseInt(file, n_cols); cout << "需要转换的图片数为: " << ImagesMat.size() << endl; for (int i = 0; i < ImagesMat.size(); ++i) { for (int r = 0; r < n_rows; ++r) { for (int c = 0; c < n_cols; ++c) { char tmp = ImagesMat[i].at<char>(r, c); file.write((char*)&tmp, sizeof(tmp)); } } printf("图片正在转换，请稍等......%.2lf%%\r", i*100.0 / (ImagesMat.size() - 1)); } printf("\n******转换完成!******\n\n"); } else cout << subName << "文件创建失败." << endl << endl; if (file.is_open()) file.close(); return; } void Image2BinaryData::Label2BinaryFile(string filefolder, vector<int>& img_Labels) { ofstream file(filefolder, ios::binary); int idx = filefolder.find_last_of("\\") + 1; string subName = filefolder.substr(idx); if (file.is_open()) { cout << subName << "文件创建成功." << endl; int magic_number = 2049; int number_of_images = img_Labels.size(); ReverseInt(file, magic_number); ReverseInt(file, number_of_images); cout << "需要转换的标签数为: " << img_Labels.size() << endl; for (int i = 0; i < img_Labels.size(); ++i) { char tmp = (char)img_Labels[i]; file.write((char*)&tmp, sizeof(tmp)); printf("标签正在转换，请稍等......%.2lf%%\r", i*100.0 / (img_Labels.size() - 1)); } printf("\n******转换完成!******\n"); } else cout << subName << "文件创建失败." << endl; if (file.is_open()) file.close(); return; }