關(guān)于tf.TFRecordReader()函數(shù)的用法解析
更新時間：2020年02月17日 15:48:21 作者：我是管小亮
今天小編就為大家分享一篇關(guān)于tf.TFRecordReader()函數(shù)的用法解析，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
讀取tfrecord數(shù)據(jù)
從TFRecords文件中讀取數(shù)據(jù)， 首先需要用tf.train.string_input_producer生成一個解析隊列。之后調(diào)用tf.TFRecordReader的tf.parse_single_example解析器。
如下圖：
解析器首先讀取解析隊列，返回serialized_example對象，之后調(diào)用tf.parse_single_example操作將Example協(xié)議緩沖區(qū)(protocol buffer)解析為張量。
簡單來說，一旦生成了TFRecords文件，接下來就可以使用隊列（queue）讀取數(shù)據(jù)了。
def read_and_decode(filename):
#根據(jù)文件名生成一個隊列
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讀取格式時的代碼：
這部分只要使用對應(yīng)的代碼就可以，主要是知道咋回事。
以上這篇關(guān)于tf.TFRecordReader()函數(shù)的用法解析就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要為大家詳細介紹了Python實現(xiàn)自動上京東搶手機的方法，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-02-02
對python中使用requests模塊參數(shù)編碼的不同處理方法
今天小編就為大家分享一篇對python中使用requests模塊參數(shù)編碼的不同處理方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05
這篇文章主要介紹了Python -m參數(shù)原理及使用方法解析,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-08-08
python數(shù)據(jù)擬合之scipy.optimize.curve_fit解讀
這篇文章主要介紹了python數(shù)據(jù)擬合之scipy.optimize.curve_fit解讀，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-12-12
下面小編就為大家?guī)硪黄獪\談python新手中常見的疑惑及解答。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2016-06-06
Python基礎(chǔ)之字符串操作常用函數(shù)集合
這篇文章主要介紹了Python基礎(chǔ)之字符串操作常用函數(shù)集合,需要的朋友可以參考下2020-02-02
在本篇文章里小編給大家分享的是關(guān)于python中subprocess實例用法及知識點詳解內(nèi)容，有需要的朋友們可以跟著學(xué)習(xí)下。2021-10-10
pycharm下打開、執(zhí)行并調(diào)試scrapy爬蟲程序的方法
本篇文章主要介紹了pycharm下打開、執(zhí)行并調(diào)試scrapy爬蟲程序的方法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-11-11
Scrapy是一個使用Python編寫的，輕量級的，簡單輕巧，并且使用起來非常的方便。使用Scrapy可以很方便的完成網(wǎng)上數(shù)據(jù)的采集工作，它為我們完成了大量的工作，而不需要自己費大力氣去開發(fā)。本文介紹了使用Python的Scrapy框架十分鐘爬取美女圖,需要的朋友可以參考下2016-12-12
關(guān)于Python面向?qū)ο缶幊痰闹R點總結(jié)
Python從設(shè)計之初就已經(jīng)是一門面向?qū)ο蟮恼Z言，正因為如此，在Python中創(chuàng)建一個類和對象是很容易的。下面這篇文章將詳細給大家介紹關(guān)于Python面向?qū)ο缶幊痰闹R點，需要的朋友可以參考借鑒，下面來一起看看吧。2017-02-02