模仿CIFAR-10数据集制作自己的数据集（亲测有效）

ゝ一纸荒年。 2022-06-12 09:41 206阅读 0赞

本代码参考博文[http://blog.csdn.net/yhl\_leo/article/details/50801226][http_blog.csdn.net_yhl_leo_article_details_50801226]，并在其基础上做了稍微的修改，vs2015上测试能够编译运行通过，并添加了自己的注释。注意，此代码需要配置opencv，我的运行环境为win10+VS2015+Opencv3.1.0.

关于如何配置Opencv3.1.0，可参考[这里][Link 1]

代码一共包含三个文件

*  BinaryDataSet.h
 *  BinaryDataSet.cpp
 *  Main.cpp

下面给出Main.cpp的代码

#include "BinaryDataSet.h"
    
    void main()
    {
    	string filefolder = "C:\\Users\\lyf\\Desktop\\test";		//文件的绝对路径
    	BinaryDataSet binData(256, 256);				//在此传入图片的大小（高，宽）
    	vector<string> fileLists = binData.getFileLists(filefolder);	//获得文件名列表
    	for (auto x : fileLists)					//显示文件名称列表
    		cout << x << endl;
    	const int size_list = fileLists.size();				//文件个数
    	cout << "Image Counts: " << size_list << endl;
    	vector<int> image_labels(size_list);				//生成标签列表
    
    	string binfile = "C:\\Users\\lyf\\Desktop\\test.bin";			//设置二进制文件保存的路径及名称
    	binData.images2Binaryfiles(filefolder, fileLists, image_labels, binfile);//将图片写入二进制文件
    	binData.readImage(binfile);		//验证
    }

其它文件中的代码可根据自己的数据集自行进行修改，代码注释是按照自己理解添加的，有些地方表述可能不清楚，请见谅。

代码下载链接:[点击链接下载][Link 2]

[http_blog.csdn.net_yhl_leo_article_details_50801226]: http://blog.csdn.net/yhl_leo/article/details/50801226
[Link 1]: http://blog.csdn.net/yf_li123/article/details/75463403
[Link 2]: http://download.csdn.net/detail/yf_li123/9906259