关于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图算法实例分析
Aug 13 Python
pycharm中连接mysql数据库的步骤详解
May 02 Python
wxPython之解决闪烁的问题
Jan 15 Python
Python实现的生产者、消费者问题完整实例
May 30 Python
python 获取毫秒数,计算调用时长的方法
Feb 20 Python
详解js文件通过python访问数据库方法
Mar 03 Python
python面试题Python2.x和Python3.x的区别
May 28 Python
PyQt5笔记之弹出窗口大全
Jun 20 Python
python 使用socket传输图片视频等文件的实现方式
Aug 07 Python
Python logging模块进行封装实现原理解析
Aug 07 Python
如何基于matlab相机标定导出xml文件
Nov 02 Python
Python Pandas pandas.read_sql_query函数实例用法分析
Jun 21 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
通过Email发送PHP错误的方法
2015/07/20 PHP
PHP下SSL加密解密、验证、签名方法(很简单)
2020/06/28 PHP
thinkphp实现分页显示功能
2016/12/03 PHP
php插件Xajax使用方法详解
2017/08/31 PHP
PHP下载大文件失败并限制下载速度的实例代码
2019/05/10 PHP
FCK调用方法..
2006/12/21 Javascript
jquery实现对联广告的方法
2015/02/05 Javascript
jQuery ajax全局函数处理session过期后的ajax跳转问题
2016/06/03 Javascript
JS动态加载脚本并执行回调操作
2016/08/24 Javascript
Bootstrap select多选下拉框实现代码
2016/12/23 Javascript
vue中七牛插件使用的实例代码
2017/07/28 Javascript
微信小程序canvas.drawImage完全显示图片问题的解决
2018/11/30 Javascript
video.js添加自定义组件的方法
2020/12/09 Javascript
[01:06]欢迎来到上海,TI9
2018/08/26 DOTA
python定时检查启动某个exe程序适合检测exe是否挂了
2013/01/21 Python
python实现将文本转换成语音的方法
2015/05/28 Python
好用的Python编辑器WingIDE的使用经验总结
2016/08/31 Python
Python 通过调用接口获取公交信息的实例
2018/12/17 Python
Python自动化运维之Ansible定义主机与组规则操作详解
2019/06/13 Python
django中使用POST方法获取POST数据
2019/08/20 Python
python 爬虫 实现增量去重和定时爬取实例
2020/02/28 Python
为什么python比较流行
2020/06/19 Python
Python .py生成.pyd文件并打包.exe 的注意事项说明
2021/03/04 Python
德国排名第一的主题公园门票网站:Attraction Tickets Direct
2019/09/09 全球购物
小区门卫工作职责
2013/12/14 职场文书
证婚人经典证婚词
2014/01/09 职场文书
会计员岗位职责
2014/03/15 职场文书
股份合作协议书
2014/04/12 职场文书
党员群众路线对照检查材料
2014/08/31 职场文书
第二批党的群众路线教育实践活动个人整改方案
2014/10/31 职场文书
鼋头渚导游词
2015/02/05 职场文书
索赔员岗位职责
2015/02/15 职场文书
Nginx + consul + upsync 完成动态负载均衡的方法详解
2021/03/31 Servers
php7中停止php-fpm服务的方法详解
2021/05/09 PHP
解决pytorch-gpu 安装失败的记录
2021/05/24 Python
SpringBoot 拦截器妙用你真的了解吗
2021/07/01 Java/Android