关于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版微信红包分配算法
May 04 Python
浅谈Python的文件类型
May 30 Python
python读取中文txt文本的方法
Apr 12 Python
对TensorFlow中的variables_to_restore函数详解
Jul 30 Python
对python 自定义协议的方法详解
Feb 13 Python
Python 实现输入任意多个数,并计算其平均值的例子
Jul 16 Python
Python 调用 Outlook 发送邮件过程解析
Aug 08 Python
pycharm配置git(图文教程)
Aug 16 Python
python实现拉普拉斯特征图降维示例
Nov 25 Python
Python3实现将一维数组按标准长度分隔为二维数组
Nov 29 Python
Python基本类型的连接组合和互相转换方式(13种)
Dec 16 Python
python sitk.show()与imageJ结合使用常见的问题
Apr 20 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类
2006/11/25 PHP
PHP扩展模块memcached长连接使用方法分析
2014/12/24 PHP
读jQuery之一(对象的组成)
2011/06/11 Javascript
利用webqq协议使用python登录qq发消息源码参考
2013/04/08 Javascript
jquery实现简单的拖拽效果实例兼容所有主流浏览器(优化篇)
2013/06/28 Javascript
javascript break指定标签打破多层循环示例
2014/01/20 Javascript
Jquery实现点击按钮,连续地向textarea中添加值的实例代码
2014/03/08 Javascript
三种检测iPhone/iPad设备方向的方法
2014/04/23 Javascript
jQuery学习笔记之基础中的基础
2015/01/19 Javascript
JavaScript中继承用法实例分析
2015/05/16 Javascript
使用Meteor配合Node.js编写实时聊天应用的范例
2015/06/23 Javascript
Jq通过td获取同行其它列td的方法
2016/10/05 Javascript
微信小程序 常见问题总结(4058,40013)及解决办法
2017/01/11 Javascript
vue.js删除动态绑定的radio的指定项
2017/06/02 Javascript
原生JS写Ajax的请求函数功能
2017/12/22 Javascript
利用jsonp解决js读取本地json跨域的问题
2018/12/11 Javascript
Vue项目history模式下微信分享爬坑总结
2019/03/29 Javascript
VUE兄弟组件传值操作实例分析
2019/10/26 Javascript
vue解决花括号数据绑定不成功的问题
2019/10/30 Javascript
JS实现表单中点击小眼睛显示隐藏密码框中的密码
2020/04/13 Javascript
JavaScript Tab菜单实现过程解析
2020/05/13 Javascript
[02:38]2018DOTA2亚洲邀请赛赛前采访-VGJ.T
2018/04/03 DOTA
python 通过麦克风录音 生成wav文件的方法
2019/01/09 Python
windows下numpy下载与安装图文教程
2019/04/02 Python
Python 实现交换矩阵的行示例
2019/06/26 Python
Django model 中设置联合约束和联合索引的方法
2019/08/06 Python
pytorch 固定部分参数训练的方法
2019/08/17 Python
Django文件上传与下载(FileFlid)
2019/10/06 Python
基于python爬取链家二手房信息代码示例
2020/10/21 Python
详解Python中的Lock和Rlock
2021/01/26 Python
介绍一下linux的文件系统
2015/10/06 面试题
2014学校庆三八妇女节活动总结
2014/03/01 职场文书
廉洁教育学习材料
2014/05/19 职场文书
教师党员公开承诺事项
2014/05/28 职场文书
党员四风问题对照检查材料思想汇报
2014/09/16 职场文书
孝女彩金观后感
2015/06/10 职场文书