您的位置:首页 > 编程语言 > C语言/C++

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

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

代码下载链接:点击链接下载
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  C++ CIFAR 制作数据集