关于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 (2)
Oct 31 Python
python进阶教程之异常处理
Aug 30 Python
Python模块包中__init__.py文件功能分析
Jun 14 Python
SVM基本概念及Python实现代码
Dec 27 Python
对numpy和pandas中数组的合并和拆分详解
Apr 11 Python
Python实现K折交叉验证法的方法步骤
Jul 11 Python
python实现证件照换底功能
Aug 20 Python
Python 闭包,函数分隔作用域,nonlocal声明非局部变量操作示例
Oct 14 Python
Python 切分数组实例解析
Nov 07 Python
将python安装信息加入注册表的示例
Nov 20 Python
tensorflow 实现打印pb模型的所有节点
Jan 23 Python
Python 处理表格进行成绩排序的操作代码
Jul 26 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+DBM的同学录程序(1)
2006/10/09 PHP
php include加载文件两种方式效率比较
2010/08/08 PHP
实用PHP会员权限控制实现原理分析
2011/05/29 PHP
PHP数组及条件,循环语句学习
2012/11/11 PHP
PHP实现支持GET,POST,Multipart/form-data的HTTP请求类
2014/09/24 PHP
php恢复数组的key为数字序列的方法
2015/04/28 PHP
PHP设置images目录不充许http访问的方法
2016/11/01 PHP
Swoole扩展的6种模式深入详解
2021/03/04 PHP
jQuery 第二课 操作包装集元素代码
2010/03/14 Javascript
JS实现QQ图片一闪一闪的效果小例子
2013/07/31 Javascript
css与javascript跨浏览器兼容性总结
2014/09/15 Javascript
jQuery.deferred对象使用详解
2016/03/18 Javascript
Extjs4.0 ComboBox如何实现三级联动
2016/05/11 Javascript
通过构造函数实例化对象的方法
2017/06/28 Javascript
vue项目中api接口管理总结
2018/04/20 Javascript
解决vue动态路由异步加载import组件,加载不到module的问题
2020/07/26 Javascript
[00:42]《辉夜杯》—职业组预选赛12月3日15点 正式打响
2015/12/03 DOTA
[01:25]DOTA2自定义游戏灵园鬼域等你踏足
2015/10/30 DOTA
Python的ORM框架SQLAlchemy入门教程
2014/04/28 Python
Python随手笔记之标准类型内建函数
2015/12/02 Python
Python读取txt某几列绘图的方法
2018/10/14 Python
python去掉 unicode 字符串前面的u方法
2018/10/21 Python
对python的unittest架构公共参数token提取方法详解
2018/12/17 Python
浅谈Python3识别判断图片主要颜色并和颜色库进行对比的方法
2019/10/25 Python
Python自动化xpath实现自动抢票抢货
2020/09/19 Python
Python中用xlwt制作表格实例讲解
2020/11/05 Python
FILA斐乐中国官方商城:意大利运动品牌
2017/01/25 全球购物
波兰运动鞋网上商店:e-Sporting
2018/02/16 全球购物
PHP如何对用户密码进行加密
2014/07/31 面试题
出生医学证明书
2014/09/15 职场文书
2014年教师思想工作总结
2014/12/03 职场文书
企业安全生产检查制度
2015/08/06 职场文书
医学生自荐信范文(2016精选篇)
2016/01/28 职场文书
PHP策略模式写法
2021/04/01 PHP
MySQL主从搭建(多主一从)的实现思路与步骤
2021/05/13 MySQL
详解OpenCV曝光融合
2022/04/29 Python