博客
关于我
C++基于文件流和armadillo读取mnist
阅读量:530 次
发布时间:2019-03-08

本文共 2548 字,大约阅读时间需要 8 分钟。

发现网上大把都是用python读取mnist的,用C++大都是用opencv读取的,但我不怎么用opencv,因此自己摸索了个使用文件流读取mnist的方法,armadillo仅作为储存矩阵的一种方式。

1. mnist文件

首先避坑,这些文件要解压。

没有肾么描述
官网截图可知,文件头很简单,只有若干个32位整数,MSB,像素和标签均是无符号字节(即unsigned char)可以先读取文件头,再读取剩下的部分。

2. 读取文件头

我觉得没什么必要啊,直接跳过不行吗

文件头都是32位,那就整四个unsigned char呗。

FILE *File = fopen(fileName, "r");	fseek(File, 0, 0);	uchar a[4];	fread(a, 4, 1, File);

这样a字符串就保存了一个整数。

x = ((((a[0] * 256) + a[1]) * 256) + a[2]) * 256 + a[3];

然后就得到了呗。

看每个文件有多少文件头,就操作几次(并可以顺便与官方的magic number进行对比),剩下的就是文件的内容了。

3. 读取内容

这部分可以依照之前的方法,一次读取一个字符,再保存至矩阵当中。例如:

uchar a;mat image(28, 28, fill::zeros); // 这是个矩阵!for(int i = 0; i < 28; i++) //28行28列的图像懒得改了	for(int j = 0; j < 28; j++)	{   		fread(a, 1, 1, File);		image(i, j) = double(a);	}

这样就读取了一张图片。其余以此类推吧。

4. 完整代码

可以复制,可以修改,也可以用于商业和学术,但是请标注原作者(就是我)。

mnist.h

#ifndef MNIST_H  #define MNIST_H#include
#include
#include
#include
#include
#define uchar unsigned charusing namespace std;using namespace arma;//小端存储转换int reverseInt(uchar *a);//读取image数据集信息mat read_mnist_image(const char *fileName);//读取label数据集信息mat read_mnist_label(const char* fileName);#endif

mnist.cpp

//mnist.cpp//作者:C艹#include "mnist.h"int reverseInt(uchar *a){   	return ((((a[0] * 256) + a[1]) * 256) + a[2]) * 256 + a[3];}mat read_mnist_image(const char *fileName){   	FILE *File = fopen(fileName, "r");	fseek(File, 0, 0);	mat image;	uchar a[4];	fread(a, 4, 1, File);	int magic = reverseInt(a);	if (magic != 2051) //magic number wrong	{   		cout << magic;		return mat(0, 0, fill::zeros);	}	fread(a, 4, 1, File);	const int num_img = reverseInt(a);	fread(a, 4, 1, File);	const int num_row = reverseInt(a);	fread(a, 4, 1, File);	const int num_col = reverseInt(a);	const int size_img = num_col * num_row;	// 文件头读取完毕	image.reshape(num_img, size_img);	uchar img[784];	for (int i = 0; i < num_img; i++)	{   		fseek(File, i*784+16, SEEK_SET);		fread(img, size_img, 1, File);		for (int j = 0; j < size_img; j++)		{   			image(i, j) = double(img[j])/256;		}	}	fclose(File);	return image;}mat read_mnist_label(const char *fileName){   	FILE* File = fopen(fileName, "r");	fseek(File, 0, 0);	uchar a[4];	fread(a, 4, 1, File);	int magic = reverseInt(a);	if (magic != 2049) //magic number wrong	{   		cout << magic;		return mat(0, 0, fill::zeros);	}	fread(a, 4, 1, File);	const int num_lab = reverseInt(a);	// 文件头读取完毕	mat label(num_lab, 10, fill::zeros);	uchar lab[1];	for (int i = 0; i < num_lab; i++)	{   		fread(lab, 1, 1, File);		label(i, int(lab[0])) = 1;	}	fclose(File);	return label;}

转载地址:http://fgwiz.baihongyu.com/

你可能感兴趣的文章
multivariate_normal TypeError: ufunc ‘add‘ output (typecode ‘O‘) could not be coerced to provided……
查看>>
MySQL DBA 数据库优化策略
查看>>
multi_index_container
查看>>
mutiplemap 总结
查看>>
MySQL Error Handling in Stored Procedures---转载
查看>>
MVC 区域功能
查看>>
MySQL FEDERATED 提示
查看>>
mysql generic安装_MySQL 5.6 Generic Binary安装与配置_MySQL
查看>>
Mysql group by
查看>>
MySQL I 有福啦,窗口函数大大提高了取数的效率!
查看>>
mysql id自动增长 初始值 Mysql重置auto_increment初始值
查看>>
MySQL in 太多过慢的 3 种解决方案
查看>>
Mysql Innodb 锁机制
查看>>
MySQL InnoDB中意向锁的作用及原理探
查看>>
MySQL InnoDB事务隔离级别与锁机制深入解析
查看>>
Mysql InnoDB存储引擎 —— 数据页
查看>>
Mysql InnoDB存储引擎中的checkpoint技术
查看>>
Mysql InnoDB存储引擎中缓冲池Buffer Pool、Redo Log、Bin Log、Undo Log、Channge Buffer
查看>>
MySQL InnoDB引擎的锁机制详解
查看>>
Mysql INNODB引擎行锁的3种算法 Record Lock Next-Key Lock Grap Lock
查看>>