關(guān)于tf.TFRecordReader()函數(shù)的用法解析

更新時(shí)間：2020年02月17日 15:48:21 作者：我是管小亮

今天小編就為大家分享一篇關(guān)于tf.TFRecordReader()函數(shù)的用法解析，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧

讀取tfrecord數(shù)據(jù)

從TFRecords文件中讀取數(shù)據(jù)，首先需要用tf.train.string_input_producer生成一個(gè)解析隊(duì)列。之后調(diào)用tf.TFRecordReader的tf.parse_single_example解析器。

如下圖：

解析器首先讀取解析隊(duì)列，返回serialized_example對(duì)象，之后調(diào)用tf.parse_single_example操作將Example協(xié)議緩沖區(qū)(protocol buffer)解析為張量。

簡單來說，一旦生成了TFRecords文件，接下來就可以使用隊(duì)列（queue）讀取數(shù)據(jù)了。

def read_and_decode(filename):
  #根據(jù)文件名生成一個(gè)隊(duì)列
  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讀取格式時(shí)的代碼：

這部分只要使用對(duì)應(yīng)的代碼就可以，主要是知道咋回事。

以上這篇關(guān)于tf.TFRecordReader()函數(shù)的用法解析就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Python實(shí)現(xiàn)自動(dòng)上京東搶手機(jī)
這篇文章主要為大家詳細(xì)介紹了Python實(shí)現(xiàn)自動(dòng)上京東搶手機(jī)的方法，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
2018-02-02
對(duì)python中使用requests模塊參數(shù)編碼的不同處理方法
今天小編就為大家分享一篇對(duì)python中使用requests模塊參數(shù)編碼的不同處理方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
2018-05-05
Python -m參數(shù)原理及使用方法解析
這篇文章主要介紹了Python -m參數(shù)原理及使用方法解析,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下
2020-08-08
python數(shù)據(jù)擬合之scipy.optimize.curve_fit解讀
這篇文章主要介紹了python數(shù)據(jù)擬合之scipy.optimize.curve_fit解讀，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教
2022-12-12
淺談python新手中常見的疑惑及解答
下面小編就為大家?guī)硪黄獪\談python新手中常見的疑惑及解答。小編覺得挺不錯(cuò)的，現(xiàn)在就分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧
2016-06-06
Python基礎(chǔ)之字符串操作常用函數(shù)集合
這篇文章主要介紹了Python基礎(chǔ)之字符串操作常用函數(shù)集合,需要的朋友可以參考下
2020-02-02
python中subprocess實(shí)例用法及知識(shí)點(diǎn)詳解
在本篇文章里小編給大家分享的是關(guān)于python中subprocess實(shí)例用法及知識(shí)點(diǎn)詳解內(nèi)容，有需要的朋友們可以跟著學(xué)習(xí)下。
2021-10-10
pycharm下打開、執(zhí)行并調(diào)試scrapy爬蟲程序的方法
本篇文章主要介紹了pycharm下打開、執(zhí)行并調(diào)試scrapy爬蟲程序的方法，小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧
2017-11-11
使用Python的Scrapy框架十分鐘爬取美女圖
Scrapy是一個(gè)使用Python編寫的，輕量級(jí)的，簡單輕巧，并且使用起來非常的方便。使用Scrapy可以很方便的完成網(wǎng)上數(shù)據(jù)的采集工作，它為我們完成了大量的工作，而不需要自己費(fèi)大力氣去開發(fā)。本文介紹了使用Python的Scrapy框架十分鐘爬取美女圖,需要的朋友可以參考下
2016-12-12
關(guān)于Python面向?qū)ο缶幊痰闹R(shí)點(diǎn)總結(jié)
Python從設(shè)計(jì)之初就已經(jīng)是一門面向?qū)ο蟮恼Z言，正因?yàn)槿绱?，在Python中創(chuàng)建一個(gè)類和對(duì)象是很容易的。下面這篇文章將詳細(xì)給大家介紹關(guān)于Python面向?qū)ο缶幊痰闹R(shí)點(diǎn)，需要的朋友可以參考借鑒，下面來一起看看吧。
2017-02-02