关于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 24 Python
详解Python的Twisted框架中reactor事件管理器的用法
May 25 Python
浅谈python字典多键值及重复键值的使用
Nov 04 Python
pandas将DataFrame的列变成行索引的方法
Apr 10 Python
python中退出多层循环的方法
Nov 27 Python
详解Python正则表达式re模块
Mar 19 Python
python手写均值滤波
Feb 19 Python
Python如何把Spark数据写入ElasticSearch
Apr 18 Python
python编写一个会算账的脚本的示例代码
Jun 02 Python
通过cmd进入python的步骤
Jun 16 Python
python爬取”顶点小说网“《纯阳剑尊》的示例代码
Oct 16 Python
使用Python实现NBA球员数据查询小程序功能
Nov 09 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 Hash函数,增强密码安全
2011/02/25 PHP
PHP+Ajax实现验证码的实时验证
2016/07/20 PHP
理解Javascript_13_执行模型详解
2010/10/20 Javascript
Javascript学习笔记-详解in运算符
2011/09/13 Javascript
javascript手工制作悬浮菜单
2015/02/12 Javascript
浅谈JavaScript中null和undefined
2015/07/09 Javascript
跟我学习javascript的arguments对象
2015/11/16 Javascript
JS实现title标题栏文字不间断滚动显示效果
2016/09/07 Javascript
Javascript 引擎工作机制详解
2016/11/30 Javascript
AngularJS中run方法的巧妙运用
2017/01/04 Javascript
javascript DOM的详解及实例代码
2017/03/06 Javascript
JS去掉字符串末尾的标点符号及删除最后一个字符的方法
2017/10/24 Javascript
js 获取json数组里面数组的长度实例
2017/10/31 Javascript
微信小程序基于本地缓存实现点赞功能的方法
2017/12/18 Javascript
浅谈React中的元素、组件、实例和节点
2018/02/27 Javascript
React BootStrap用户体验框架快速上手
2018/03/06 Javascript
vue中的过滤器及其时间格式化问题
2020/04/09 Javascript
javascript设计模式 ? 策略模式原理与用法实例分析
2020/04/21 Javascript
[00:23]DOTA2群星共贺开放测试 25日无码时代来袭
2013/09/23 DOTA
Python中生成器和yield语句的用法详解
2015/04/17 Python
一些常用的Python爬虫技巧汇总
2016/09/28 Python
利用python实现xml与数据库读取转换的方法
2017/06/17 Python
Python图像处理之直线和曲线的拟合与绘制【curve_fit()应用】
2018/12/26 Python
对Python 简单串口收发GUI界面的实例详解
2019/06/12 Python
python用类实现文章敏感词的过滤方法示例
2019/10/27 Python
python 多进程队列数据处理详解
2019/12/23 Python
Python3 assert断言实现原理解析
2020/03/02 Python
keras读取训练好的模型参数并把参数赋值给其它模型详解
2020/06/15 Python
python如何将图片转换素描画
2020/09/08 Python
优质飞蝇钓和渔具:RiverBum
2020/05/10 全球购物
乌克兰巴士票购买网站:inBus
2021/03/12 全球购物
医学专业个人求职自荐信格式
2013/09/23 职场文书
大专生自我鉴定范文
2013/10/01 职场文书
最新奶茶店创业计划书
2014/01/25 职场文书
中专生自荐信
2014/06/25 职场文书
法人委托书
2014/07/31 职场文书