TF:利用TF读取数据操作,将CIFAR-10 数据集中的训练图片读取出来,并保存为.jpg格式

简介: TF:利用TF读取数据操作,将CIFAR-10 数据集中的训练图片读取出来,并保存为.jpg格式

输出结果

image.png



核心代码


def inputs_origin(data_dir):

 filenames = [os.path.join(data_dir, 'data_batch_%d.bin' % i)

              for i in range(1, 6)]

……

 filename_queue = tf.train.string_input_producer(filenames)

 read_input = cifar10_input.read_cifar10(filename_queue)    

 reshaped_image = tf.cast(read_input.uint8image, tf.float32)

 

 return reshaped_image  

if __name__ == '__main__':

 with tf.Session() as sess:

   reshaped_image = inputs_origin('cifar10_data/cifar-10-batches-bin')

   threads = tf.train.start_queue_runners(sess=sess)

   sess.run(tf.global_variables_initializer())

   if not os.path.exists('cifar10_data/raw/'):

     os.makedirs('cifar10_data/raw/')

   for i in range(30):

     image_array = sess.run(reshaped_image)

     scipy.misc.toimage(image_array).save('cifar10_data/raw/%d.jpg' % i)


 


相关文章
|
6月前
|
数据处理 开发工具 git
coco2017数据集转换为yolo格式(记录过程)
最近做一个yolov5的落地应用项目,用的anylabeling打标,需要将coco2017的数据集转为yolo格式,故写下记录过程!
|
TensorFlow 算法框架/工具
【tensorflow】TF1.x保存与读取.pb模型写法介绍
由于TF里面的概念比较接地气,所以用tf1.x保存.pb模型时总是怕有什么操作漏掉了,会造成保存的模型是缺少变量数据或者没有保存图,所以先明确一下:用TF1.x保存模型时只需要保存模型的输入输出的变量(多输入就保存多个),不需要保存中间的变量;用TF1.x加载模型时只需要加载保存的模型,然后读一下输入输出变量(多输入就读多个),不需要初始化(反而会重置掉变量的值)。
153 0
|
1月前
|
XML JSON 数据可视化
数据集学习笔记(二): 转换不同类型的数据集用于模型训练(XML、VOC、YOLO、COCO、JSON、PNG)
本文详细介绍了不同数据集格式之间的转换方法,包括YOLO、VOC、COCO、JSON、TXT和PNG等格式,以及如何可视化验证数据集。
64 1
数据集学习笔记(二): 转换不同类型的数据集用于模型训练(XML、VOC、YOLO、COCO、JSON、PNG)
|
1月前
|
XML 计算机视觉 数据格式
数据集学习笔记(四):VOC转COCO数据集并据txt中图片的名字批量提取对应的图片并保存到另一个文件夹
这篇文章介绍了如何将VOC数据集转换为COCO数据集的格式,并通过Python脚本根据txt文件中列出的图片名称批量提取对应的图片并保存到另一个文件夹。
20 3
|
3月前
|
TensorFlow 算法框架/工具 索引
mnist 数据集读取
【8月更文挑战第9天】mnist 数据集读取。
35 3
|
PyTorch 算法框架/工具
如何将x_data和y_data利用torch转换成小批量数据,并要求打乱数据,如何处理?
首先,使用TensorDataset将x_data和y_data合并到一个数据集中。然后,使用DataLoader创建一个迭代器,以便逐个处理每个小批量数据。在这里,batch_size设置为2,这意味着每个小批量将包含2个样本。shuffle参数设置为True,表示要对数据进行随机打乱。在遍历每个小批量数据时,可以在循环体内执行训练或评估操作。
182 0
|
算法框架/工具 Caffe
Caffe:使用 classify.py 批量对图片分类
一般使用 Caffe 训练完网络后,会用 `test.bin` 来测试一下网络的精度,然后还能用 `classification.bin` 来用网络对图片进行单张的分类,但是一张一张的分,效率很低,所以我改写了 `classify.py` 文件,使其读取 test.txt 文件批量分类,输出具体哪一张图片分错了。
131 0
|
缓存 NoSQL MongoDB
TensorFlow2.0(10):加载自定义图片数据集到Dataset
TensorFlow2.0(10):加载自定义图片数据集到Dataset
C#编程-66:DataSet数据集的简单操作
C#编程-66:DataSet数据集的简单操作
100 0
C#编程-66:DataSet数据集的简单操作
ML之FE:数据随机抽样之利用pandas的sample函数对超大样本的数据集进行随机采样,并另存为csv文件
ML之FE:数据随机抽样之利用pandas的sample函数对超大样本的数据集进行随机采样,并另存为csv文件