关于tf.TFRecordReader()函数的用法解析
更新时间：2020年02月17日 15:48:21 作者：我是管小亮
今天小编就为大家分享一篇关于tf.TFRecordReader()函数的用法解析，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
读取tfrecord数据
从TFRecords文件中读取数据， 首先需要用tf.train.string_input_producer生成一个解析队列。之后调用tf.TFRecordReader的tf.parse_single_example解析器。
如下图：
解析器首先读取解析队列，返回serialized_example对象，之后调用tf.parse_single_example操作将Example协议缓冲区(protocol buffer)解析为张量。
简单来说，一旦生成了TFRecords文件，接下来就可以使用队列（queue）读取数据了。
def read_and_decode(filename):
#根据文件名生成一个队列
filename_queue = tf.train.string_input_producer([filename])
reader = tf.TFRecordReader()
_, serialized_example = reader.read(filename_queue) #返回文件名和文件
features = tf.parse_single_example(serialized_example,
features={
'label': tf.FixedLenFeature([], tf.int64),
'img_raw' : tf.FixedLenFeature([], tf.string),
})
img = tf.decode_raw(features['img_raw'], tf.uint8)
img = tf.reshape(img, [224, 224, 3])
img = tf.cast(img, tf.float32) * (1. / 255) - 0.5
label = tf.cast(features['label'], tf.int32)
return img, label
举例：
下面代码是我的程序中利用TFRecord读取格式时的代码：
这部分只要使用对应的代码就可以，主要是知道咋回事。
以上这篇关于tf.TFRecordReader()函数的用法解析就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要为大家详细介绍了Python实现自动上京东抢手机的方法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-02-02
对python中使用requests模块参数编码的不同处理方法
今天小编就为大家分享一篇对python中使用requests模块参数编码的不同处理方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
这篇文章主要介绍了Python -m参数原理及使用方法解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-08-08
python数据拟合之scipy.optimize.curve_fit解读
这篇文章主要介绍了python数据拟合之scipy.optimize.curve_fit解读，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-12-12
下面小编就为大家带来一篇浅谈python新手中常见的疑惑及解答。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2016-06-06
这篇文章主要介绍了Python基础之字符串操作常用函数集合,需要的朋友可以参考下2020-02-02
在本篇文章里小编给大家分享的是关于python中subprocess实例用法及知识点详解内容，有需要的朋友们可以跟着学习下。2021-10-10
本篇文章主要介绍了pycharm下打开、执行并调试scrapy爬虫程序的方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-11-11
Scrapy是一个使用Python编写的，轻量级的，简单轻巧，并且使用起来非常的方便。使用Scrapy可以很方便的完成网上数据的采集工作，它为我们完成了大量的工作，而不需要自己费大力气去开发。本文介绍了使用Python的Scrapy框架十分钟爬取美女图,需要的朋友可以参考下2016-12-12
Python从设计之初就已经是一门面向对象的语言，正因为如此，在Python中创建一个类和对象是很容易的。下面这篇文章将详细给大家介绍关于Python面向对象编程的知识点，需要的朋友可以参考借鉴，下面来一起看看吧。2017-02-02
最新评论