關(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虛擬環(huán)境完美部署教程，非常不錯，具有一定的參考借鑒價值,需要的朋友可以參考下2019-08-08
python通過floor函數(shù)舍棄小數(shù)位的方法
這篇文章主要介紹了python通過floor函數(shù)舍棄小數(shù)位的方法,實(shí)例分析了Python中floor函數(shù)的功能及使用技巧,需要的朋友可以參考下2015-03-03
這篇文章主要為大家介紹了Python如何優(yōu)雅的重試詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-07-07
使用Python實(shí)現(xiàn)屏幕截圖的兩種方法
Python作為一種高效的編程語言,可以通過一些庫來實(shí)現(xiàn)對屏幕的截圖操作,本文主要介紹了使用Python實(shí)現(xiàn)屏幕截圖的兩種方法,具有一定的 參考價值,感興趣的可以了解一下2023-12-12
這篇文章主要介紹了python 基于opencv操作攝像頭的方法，幫助大家更好的理解和使用python，感興趣的朋友可以了解下2020-12-12
python入門之基礎(chǔ)語法學(xué)習(xí)筆記
學(xué)習(xí)python過程中需要了解的一些基礎(chǔ)語法特整理一下方便更開始接觸python的朋友2020-02-02
python庫構(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文件,通過指定構(gòu)建工具如setuptools或poetry,管理項(xiàng)目依賴,配置工具行為等,需要的朋友可以參考下2024-09-09
Python實(shí)現(xiàn)刪除文件但保留指定文件
這篇文章主要介紹了Python實(shí)現(xiàn)刪除文件但保留指定文件,本文直接給出實(shí)現(xiàn)代碼,并同時給出代碼解釋,需要的朋友可以參考下2015-06-06
Python中常用于時間的模塊有time、datetime 和 calendar，顧名思義 time 是表示時間（時、分、秒、毫秒）等，calendar 是表示日歷時間的，本章先討論 datetime 模塊,需要的朋友可以參考下2022-06-06
numpy數(shù)組之讀寫文件的實(shí)現(xiàn)
本文主要介紹了numpy數(shù)組之讀寫文件的實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-02-02