关于tf.TFRecordReader()函数的用法解析


Posted in Python onFebruary 17, 2020

读取tfrecord数据

从TFRecords文件中读取数据, 首先需要用tf.train.string_input_producer生成一个解析队列。之后调用tf.TFRecordReader的tf.parse_single_example解析器。

如下图:

关于tf.TFRecordReader()函数的用法解析

解析器首先读取解析队列,返回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()函数的用法解析

这部分只要使用对应的代码就可以,主要是知道咋回事。

以上这篇关于tf.TFRecordReader()函数的用法解析就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python实现备份文件实例
Sep 16 Python
Python中的exec、eval使用实例
Sep 23 Python
在Python的web框架中中编写日志列表的教程
Apr 30 Python
Python使用pygame模块编写俄罗斯方块游戏的代码实例
Dec 08 Python
在Python的Flask中使用WTForms表单框架的基础教程
Jun 07 Python
python leetcode 字符串相乘实例详解
Sep 03 Python
Python3爬虫全国地址信息
Jan 05 Python
详解Pandas之容易让人混淆的行选择和列选择
Jul 10 Python
TensorFlow绘制loss/accuracy曲线的实例
Jan 21 Python
基于Tensorflow:CPU性能分析
Feb 10 Python
Python实现图片查找轮廓、多边形拟合、最小外接矩形代码
Jul 14 Python
python中子类与父类的关系基础知识点
Feb 02 Python
将数据集制作成VOC数据集格式的实例
Feb 17 #Python
将labelme格式数据转化为标准的coco数据集格式方式
Feb 17 #Python
开启Django博客的RSS功能的实现方法
Feb 17 #Python
Python3打包exe代码2种方法实例解析
Feb 17 #Python
Django 博客实现简单的全文搜索的示例代码
Feb 17 #Python
Python使用qrcode二维码库生成二维码方法详解
Feb 17 #Python
django2.2 和 PyMySQL版本兼容问题
Feb 17 #Python
You might like
PHP语言中global和$GLOBALS[]的分析 之二
2012/02/02 PHP
Notice: Trying to get property of non-object problem(PHP)解决办法
2012/03/11 PHP
PHP 自动加载的简单实现(推荐)
2016/08/12 PHP
PHP精确计算功能示例
2016/11/29 PHP
ThinkPHP框架表单验证操作方法
2017/07/19 PHP
javascript下操作css的float属性的特殊写法
2007/08/22 Javascript
用JS操作FRAME中的IFRAME及其内容的实现代码
2008/07/26 Javascript
用RadioButten或CheckBox实现div的显示与隐藏
2013/09/21 Javascript
jQuery focus和blur事件的应用详解
2014/01/26 Javascript
JQuery操作元素的css样式
2015/03/09 Javascript
浅析2种JavaScript继承方式
2015/12/04 Javascript
微信小程序 欢迎页面的制作(源码下载)
2017/01/09 Javascript
JavaScript在控件上添加倒计时功能的实现代码
2017/07/04 Javascript
浅谈vue中数据双向绑定的实现原理
2017/09/14 Javascript
vue2.0 computed 计算list循环后累加值的实例
2018/03/07 Javascript
JavaScript实现一个带AI的井字棋游戏源码
2018/05/21 Javascript
移动端如何用下拉刷新的方式实现上拉加载
2018/12/10 Javascript
基于jQuery的时间戳与日期间的转化
2019/06/21 jQuery
layui多图上传实现删除功能的例子
2019/09/23 Javascript
Vue2.x-使用防抖以及节流的示例
2021/03/02 Vue.js
Python实现程序的单一实例用法分析
2015/06/03 Python
Flask框架URL管理操作示例【基于@app.route】
2018/07/23 Python
python实现求特征选择的信息增益
2018/12/18 Python
DRF跨域后端解决之django-cors-headers的使用
2019/01/27 Python
Python中Numpy ndarray的使用详解
2019/05/24 Python
python requests使用socks5的例子
2019/07/25 Python
python 使用事件对象asyncio.Event来同步协程的操作
2020/05/04 Python
使用Python将语音转换为文本的方法
2020/08/10 Python
HTML5中的postMessage API基本使用教程
2016/05/20 HTML / CSS
Java面试题:请说出如下代码的输出结果
2013/04/22 面试题
写好自荐信的技巧
2013/11/08 职场文书
毕业生面试求职信
2014/06/23 职场文书
教你用Java Swing实现自助取款机系统
2021/06/11 Java/Android
Python实现简单的俄罗斯方块游戏
2021/09/25 Python
python实现简单石头剪刀布游戏
2021/10/24 Python
yyds什么意思?90后已经听不懂00后讲话了……
2022/02/03 杂记