关于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原始套接字编程示例分享
Feb 21 Python
简单的Python抓taobao图片爬虫
Oct 26 Python
分析Python的Django框架的运行方式及处理流程
Apr 08 Python
在Python的web框架中中编写日志列表的教程
Apr 30 Python
python微信公众号之关注公众号自动回复
Oct 25 Python
python实现控制台打印的方法
Jan 12 Python
python 批量解压压缩文件的实例代码
Jun 27 Python
tensorflow通过模型文件,使用tensorboard查看其模型图Graph方式
Jan 23 Python
Python模块/包/库安装的六种方法及区别
Feb 24 Python
python实现批量命名照片
Jun 18 Python
PyQt5结合matplotlib绘图的实现示例
Sep 15 Python
python中requests模拟登录的三种方式(携带cookie/session进行请求网站)
Nov 17 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中判断一个请求是ajax请求还是普通请求的方法
2011/06/28 PHP
PHP获取指定函数定义在哪个文件中以及其所在的行号实例
2014/05/08 PHP
PHP判断密码强度的方法详解
2017/05/26 PHP
PHP设计模式之工厂模式定义与用法详解
2018/04/03 PHP
laravel框架邮箱认证实现方法详解
2019/11/22 PHP
js中根据字数截取字符串,不能截断url
2012/01/12 Javascript
精心挑选的15个jQuery下拉菜单制作教程
2012/06/15 Javascript
通过正则格式化url查询字符串实现代码
2012/12/28 Javascript
jquery中常用的函数和属性详细解析
2014/03/07 Javascript
JS使用ajax从xml文件动态获取数据显示的方法
2015/03/24 Javascript
js+HTML5实现视频截图的方法
2015/06/16 Javascript
基于jQuery实现文本框只能输入数字(小数、整数)
2016/01/14 Javascript
jquery组件WebUploader文件上传用法详解
2020/10/23 Javascript
bootstrap jquery dataTable 异步ajax刷新表格数据的实现方法
2017/02/10 Javascript
微信小程序实战之自定义toast(6)
2017/04/18 Javascript
Typescript 中的 interface 和 type 到底有什么区别详解
2019/06/18 Javascript
JavaScript this关键字指向常用情况解析
2020/09/02 Javascript
python数字图像处理实现直方图与均衡化
2018/05/04 Python
解决在pycharm中显示额外的 figure 窗口问题
2019/01/15 Python
Python图像读写方法对比
2020/11/16 Python
C++和python实现阿姆斯特朗数字查找实例代码
2020/12/07 Python
selenium+超级鹰实现模拟登录12306
2021/01/24 Python
localStorage的过期时间设置的方法详解
2018/11/26 HTML / CSS
说说你所熟悉或听说过的j2ee中的几种常用模式?及对设计模式的一些看法
2012/05/24 面试题
thinkphp5 redis缓存新增方法实例讲解
2021/03/24 PHP
房地产销售计划书
2014/01/10 职场文书
早读迟到检讨书
2014/01/24 职场文书
《翻越远方的大山》教学反思
2014/04/13 职场文书
银行金融服务方案
2014/06/11 职场文书
员工保密协议书
2014/09/27 职场文书
行风评议整改报告
2014/11/06 职场文书
通用员工手册范本
2015/05/14 职场文书
新生开学寄语大全
2015/05/28 职场文书
Python语言内置数据类型
2022/02/24 Python
python模板入门教程之flask Jinja
2022/04/11 Python
Golang 遍历二叉树
2022/04/19 Golang