關(guān)于tf.TFRecordReader()函數(shù)的用法解析
更新時(shí)間：2020年02月17日 15:48:21 作者：我是管小亮
今天小編就為大家分享一篇關(guān)于tf.TFRecordReader()函數(shù)的用法解析，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
讀取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)解析為張量。
簡(jiǎn)單來(lái)說(shuō)，一旦生成了TFRecords文件，接下來(lái)就可以使用隊(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ù)的用法解析就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了python虛擬環(huán)境完美部署教程，非常不錯(cuò)，具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2019-08-08
python通過(guò)floor函數(shù)舍棄小數(shù)位的方法
這篇文章主要介紹了python通過(guò)floor函數(shù)舍棄小數(shù)位的方法,實(shí)例分析了Python中floor函數(shù)的功能及使用技巧,需要的朋友可以參考下2015-03-03
這篇文章主要為大家介紹了Python如何優(yōu)雅的重試詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-07-07
使用Python實(shí)現(xiàn)屏幕截圖的兩種方法
Python作為一種高效的編程語(yǔ)言,可以通過(guò)一些庫(kù)來(lái)實(shí)現(xiàn)對(duì)屏幕的截圖操作,本文主要介紹了使用Python實(shí)現(xiàn)屏幕截圖的兩種方法,具有一定的 參考價(jià)值,感興趣的可以了解一下2023-12-12
這篇文章主要介紹了python 基于opencv操作攝像頭的方法，幫助大家更好的理解和使用python，感興趣的朋友可以了解下2020-12-12
python入門之基礎(chǔ)語(yǔ)法學(xué)習(xí)筆記
學(xué)習(xí)python過(guò)程中需要了解的一些基礎(chǔ)語(yǔ)法特整理一下方便更開始接觸python的朋友2020-02-02
python庫(kù)構(gòu)建之pyproject.toml配置文件詳解
pyproject.toml是Python項(xiàng)目標(biāo)準(zhǔn)化配置文件,由PEP?518引入,用于定義構(gòu)建系統(tǒng)、項(xiàng)目元數(shù)據(jù)和依賴管理,它替代了傳統(tǒng)的setup.cfg文件,通過(guò)指定構(gòu)建工具如setuptools或poetry,管理項(xiàng)目依賴,配置工具行為等,需要的朋友可以參考下2024-09-09
Python實(shí)現(xiàn)刪除文件但保留指定文件
這篇文章主要介紹了Python實(shí)現(xiàn)刪除文件但保留指定文件,本文直接給出實(shí)現(xiàn)代碼,并同時(shí)給出代碼解釋,需要的朋友可以參考下2015-06-06
Python中常用于時(shí)間的模塊有time、datetime 和 calendar，顧名思義 time 是表示時(shí)間（時(shí)、分、秒、毫秒）等，calendar 是表示日歷時(shí)間的，本章先討論 datetime 模塊,需要的朋友可以參考下2022-06-06
numpy數(shù)組之讀寫文件的實(shí)現(xiàn)
本文主要介紹了numpy數(shù)組之讀寫文件的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2023-02-02