关于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中的localtime()方法使用详解
May 22 Python
python访问mysql数据库的实现方法(2则示例)
Jan 06 Python
Python设计模式中单例模式的实现及在Tornado中的应用
Mar 02 Python
python实现媒体播放器功能
Feb 11 Python
python向已存在的excel中新增表,不覆盖原数据的实例
May 02 Python
python写入已存在的excel数据实例
May 03 Python
pandas.DataFrame.to_json按行转json的方法
Jun 05 Python
python3读取csv和xlsx文件的实例
Jun 22 Python
vue常用指令代码实例总结
Mar 16 Python
用Python 爬取猫眼电影数据分析《无名之辈》
Jul 24 Python
详解Python利用configparser对配置文件进行读写操作
Nov 03 Python
Python还能这么玩之用Python做个小游戏的外挂
Jun 04 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
ThinkPHP设置禁止百度等搜索引擎转码(简单实用)
2016/02/15 PHP
PHP截取IE浏览器并缩小原图的方法
2016/03/04 PHP
PHP 将数组打乱 shuffle函数的用法及简单实例
2016/06/17 PHP
php版微信公众号接口实现发红包的方法
2016/10/14 PHP
PHP实现的字符串匹配算法示例【sunday算法】
2017/12/19 PHP
用jquery实现点击栏目背景色改变
2012/12/10 Javascript
js实现单一html页面两套css切换代码
2013/04/11 Javascript
快速学习JavaScript的6个思维技巧
2015/10/13 Javascript
jQuery EasyUI之DataGrid使用实例详解
2016/01/04 Javascript
Easyui 之 Treegrid 笔记
2016/04/29 Javascript
js 上传文件预览的简单实例
2016/08/16 Javascript
JS中用try catch对代码运行的性能影响分析
2016/12/26 Javascript
JS中使用正则表达式g模式和非g模式的区别
2017/04/01 Javascript
学习使用Bootstrap栅格系统
2017/05/11 Javascript
基于VUE移动音乐WEBAPP跨域请求失败的解决方法
2018/01/16 Javascript
vue.js内置组件之keep-alive组件使用
2018/07/10 Javascript
微信小程序实现日期格式化和倒计时
2020/11/01 Javascript
python线程池的实现实例
2013/11/18 Python
Python向excel中写入数据的方法
2019/05/05 Python
Python 3.8正式发布重要新功能一览
2019/10/17 Python
Pytorch DataLoader 变长数据处理方式
2020/01/08 Python
浅谈keras.callbacks设置模型保存策略
2020/06/18 Python
香港No.1得奖零食网:香港零食大王
2016/07/22 全球购物
西班牙第一的网上药房:PromoFarma.com
2017/04/17 全球购物
东芝官网商城:还原日式美学,打造美好生活
2018/12/27 全球购物
法国在线购买汽车轮胎网站:123pneus.fr
2019/02/25 全球购物
工商技校毕业生自荐信
2013/11/15 职场文书
自我评价是什么
2014/01/04 职场文书
事业单位接收函
2014/01/10 职场文书
英语教师个人总结
2015/02/09 职场文书
考博导师推荐信范文
2015/03/27 职场文书
客服专员岗位职责范本
2015/04/07 职场文书
安全教育的主题班会
2015/08/13 职场文书
浅谈Redis存储数据类型及存取值方法
2021/05/08 Redis
基于MySql验证的vsftpd虚拟用户
2021/11/07 MySQL
SQL Server内存机制浅探
2022/04/06 SQL Server