码迷,mamicode.com
首页 > 其他好文 > 详细

Tensorflow机器学习入门——cifar10数据集的读取、展示与保存

时间:2020-02-15 15:33:11      阅读:64      评论:0      收藏:0      [点我收藏+]

标签:com   www   http   ima   保存数据   保存图片   href   mat   机器学习   

  • 基本信息
  • 数据集下载解压后的目录结构:
    技术图片
  • 读取、打印和保存数据集中指定的图片:
    import pickle
    import matplotlib.pyplot as plt
    
    CIFAR_DIR ="cifar10_data/cifar-10-batches-bin/data_batch_1.bin"#数据集路径
    with open(CIFAR_DIR , rb) as f:
        data = pickle.load(f, encoding=bytes)
    
    print(----------batch1的基本信息-------------)    
    print(data的数据类型:,type(data)) # 输出 <class ‘dict‘>
    print(字典的key名称:,data.keys()) # 输出 dict_keys([b‘filenames‘, b‘data‘, b‘labels‘, b‘batch_label‘])
    print(bdata的数据类型,type(data[bdata])) # 输出 <class ‘numpy.ndarray‘>
    print(bdata的数据形状,data[bdata].shape) # 输出 (10000, 3072) 说明有 10000 个样本, 3072个特征
    
    index=4#打印第几张图片
    print(-----------第%d张图片信息----------%index)
    print(filenames:,data[bfilenames][index])
    print(labels:,data[blabels][index])
    print(batch_label:,data[bbatch_label][index])
    image_arr = data[bdata][index] # 拿出 第 index 个样本
    image_arr = image_arr.reshape((3, 32, 32)) # 将一维向量改变形状得到这样一个元组:(高,宽,通道数)
    image_arr = image_arr.transpose((1, 2, 0)) 
    plt.imshow(image_arr) # 输出图片
    plt.savefig("cifar10_data/raw/%d.png"%index)#保存图片
    plt.show()
  • 打印出的图片
  • 技术图片

     

     

Tensorflow机器学习入门——cifar10数据集的读取、展示与保存

标签:com   www   http   ima   保存数据   保存图片   href   mat   机器学习   

原文地址:https://www.cnblogs.com/Fengqiao/p/cifar10_read.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!