tensorflow-TFRecord 用法

tensorflow-TFRecord 文件详解
TFRecord 是 tensorflow 内置的文件格式，它是一种二进制文件，具有以下优点：

统一各种输入文件的操作
更好的利用内存，方便复制和移动
将二进制数据和标签(label)存储在同一个文件中

import os
import numpy as np
import tempfile
import tensorflow as tf# example_path = os.path.join(tempfile.gettempdir(), "example.tfrecords")example_path='./temp.tfrecords'
np.random.seed(0)# Write the records to a file.
with tf.io.TFRecordWriter(example_path) as file_writer:for _ in range(4):#产生随机数x, y = np.random.random(), np.random.random()print(x,y,'--->')x1=tf.train.Feature(float_list=tf.train.FloatList(value=[x]))y1=tf.train.Feature(float_list=tf.train.FloatList(value=[y]))print('x1=',x1)print('y1=',y1)feature0={"x": x1,"y":y1 ,}print('feature0=',feature0)features0=tf.train.Features(feature=feature0)print('features=',feature0)record_bytes = tf.train.Example(features=features0).SerializeToString()file_writer.write(record_bytes)# Read the data back out.
def decode_fn(record_bytes):return tf.io.parse_single_example(# Datarecord_bytes,# Schema{"x": tf.io.FixedLenFeature([], dtype=tf.float32),"y": tf.io.FixedLenFeature([], dtype=tf.float32)})for batch in tf.data.TFRecordDataset([example_path]).map(decode_fn):print("x = {x:.4f},  y = {y:.4f}".format(**batch))

tensorflow-TFRecord 用法相关推荐

Tensorflow—TFRecord文件生成与读取
Tensorflow-TFRecord文件生成与读取微信公众号:幼儿园的学霸个人的学习笔记,关于OpenCV,关于机器学习, -.问题或建议,请公众号留言; 目录文章目录 Tensorflow- ...
Tensorflow—tfrecord数据集生成与使用
参考内容: 数据读取的官方教程:Tensorflow导入数据以及使用数据 tfrecord数据集生成: 数据准备:图片数据+图片目录与label一一对应的的txt 先读取图片信息的txt文件,得到每个 ...
TensorFlow TFRecord
把mnist数据集另存为TFRecord格式 import tensorflow as tf from tensorflow.examples.tutorials.mnist import input ...
python cos函数_Python Tensorflow cos()用法及代码示例
Tensorflow是Google开发的开源机器学习库.它的应用之一是开发深度神经网络. 模块tensorflow.math为许多基本的数学运算提供支持.功能tf.cos()[别名tf.math.co ...
TensorFlow TFRecord数据集的生成与显示
##TFRecord## TensorFlow提供了TFRecord的格式来统一存储数据,TFRecord格式是一种将图像数据和标签放在一起的二进制文件,能更好的利用内存,在tensorflow中快速 ...
Tensorflow TFRecord: Can't parse serialized example
问题描述 InvalidArgumentError: Key: 'label'. Can't parse serialized Example. 保存tfrecord def save_tfrecor ...
TensorFlow 笔记1--基本用法
TensorFlow 基本用法 tensorflow编程的基本框架导入所需要的包准备数据构建模型/构建图定义变量.计算op 定义优化目标及方法op 定义初始化op 启动/运行图运行初始化op ...
TensorFlow基础（1）特点安装基本用法
TensorFlow介绍 TensorFlow是一个采用数据流图(data flow graphs),用于数值计算的开源软件库.TensorFlow 最初由Google大脑小组(隶属于Google机器 ...
Tensorflow学习笔记——Summary用法
最近在研究tensorflow自带的例程speech_command,顺便学习tensorflow的一些基本用法. 其中tensorboard 作为一款可视化神器,可以说是学习tensorflow时模 ...
tensorflow op tf.global_variables_initializer
一.安装目前用了tensorflow.deeplearning4j两个深度学习框架, tensorflow 之前一直支持到python 3.5,目前以更新到3.6,故安装最新版体验使用. 慢慢长征路 ...

tensorflow-TFRecord 用法

tensorflow-TFRecord 用法相关推荐

最新文章

热门文章