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

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

关于如何配置Opencv3.1.0,可参考这里

代码一共包含三个文件

  • 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);		//验证
}
其它文件中的代码可根据自己的数据集自行进行修改,代码注释是按照自己理解添加的,有些地方表述可能不清楚,请见谅。

代码下载链接:点击链接下载


猜你喜欢

转载自blog.csdn.net/YF_Li123/article/details/75578584
今日推荐