datasets: mnist

最新推荐文章于 2024-09-06 17:58:36 发布

syqnyue

最新推荐文章于 2024-09-06 17:58:36 发布

阅读量1.8k

点赞数

分类专栏： dataset

本文链接：https://blog.csdn.net/syqnyue/article/details/85068630

版权

本文介绍了两种读取MNIST数据集的方法，包括直接处理二进制文件及获取像素值在[0, 1]范围内的图片。在使用数据集时，需要注意像素点值的范围调整。" 115093925,10552775,使用Notepad++运行PHP程序教程,"['Notepad++', 'PHP', '编程工具', '开发环境', '命令行']

摘要由CSDN通过智能技术生成

方法一：
直接读取二进制文件
像素点值的范围[0, 255]，需要return X/255.,y_vec

  def load_mnist(self):
    data_dir = os.path.join("./data", self.dataset_name)
    
    fd = open(os.path.join(data_dir,'train-images.idx3-ubyte'))
    # 转化成 numpy 数组, 像素点的范围是[0,255]
    loaded = np.fromfile(file=fd,dtype=np.uint8)
    # 根据 mnist 官网描述的数据格式，图像像素从 16 字节开始
    trX = loaded[16:].reshape((60000,28,28,1)).astype(np.float)

    fd = open(os.path.join(data_dir,'train-labels.idx1-ubyte'))
    loaded = np.fromfile(file=fd,dtype=np.uint8)
    trY = loaded[8:].reshape((60000)).astype(np.float)

    fd = open(os.path.join(data_dir,'t10k-images.idx3-ubyte'))
    loaded = np.fromfile(file=fd,dtype=np.uint8)
    teX = loaded[16:].reshape((10000,28,28,1)).astype(np.float)

    fd = open(os.path.join(data_dir,'t10k-labels.idx1-ubyte'))
    loaded = np.fromfil