tensorflow如何批量讀取圖片
本文實(shí)例為大家分享了tensorflow如何批量讀取圖片的具體代碼,供大家參考,具體內(nèi)容如下


代碼:
import tensorflow as tf
import os
def picread(filelist):
"""
讀取狗的圖片并轉(zhuǎn)換成張量
:param filelist: 文件路f徑+名字的列表
:return: 每張圖片的張量
"""
# 1.構(gòu)造文件的隊(duì)列
file_queue = tf.train.string_input_producer(filelist)
# 2.構(gòu)造閱讀器去讀取圖片內(nèi)容(默認(rèn)讀取一張圖片)
reader = tf.WholeFileReader()
key,value = reader.read(file_queue)
# 3.對(duì)讀取的圖片進(jìn)行解碼
image = tf.image.decode_jpeg(value)
# 4.處理圖片的大小(統(tǒng)一大?。?
image_resize = tf.image.resize_images(image,[200,200])
# 注意:一定要把樣本的形狀固定,在批處理中要求所有數(shù)據(jù)的形狀必須固定
image_resize.set_shape([200,200,3])
# 5.進(jìn)行批處理
image_resize_batch = tf.train.batch([image_resize],batch_size=3,num_threads=1,capacity=3)
return image_resize
#批處理大小,跟隊(duì)列,數(shù)據(jù)的數(shù)量沒(méi)有影響,只決定 這批次處理多少數(shù)據(jù)
if __name__ == "__main__":
# 1.找到文件,放入列表 路徑+名字 ->列表當(dāng)中
file_name = os.listdir("./data/dogpic/")
filelist = [os.path.join("./data/dogpic/",file) for file in file_name ]
image_batch= picread(filelist)
#開(kāi)啟會(huì)話運(yùn)行結(jié)果
with tf.Session() as sess:
#定義一個(gè)線程協(xié)調(diào)器
coord = tf.train.Coordinator()
#開(kāi)啟讀文件的線程
threads = tf.train.start_queue_runners(sess,coord=coord)
#打印讀取的內(nèi)容
print(sess.run([image_batch]))
#回收子線程
coord.request_stop()
coord.join(threads)
結(jié)果:

以上就是本文的全部?jī)?nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
基于OpenCv實(shí)現(xiàn)的人臉識(shí)別(附Python完整代碼)
人臉識(shí)別是基于人的臉部特征信息進(jìn)行身份識(shí)別的一種生物識(shí)別技術(shù),下面這篇文章主要給大家介紹了關(guān)于如何基于OpenCv實(shí)現(xiàn)的人臉識(shí)別,文中還附Python完整代碼,文中通過(guò)實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-11-11
Python的getattr函數(shù)方法學(xué)習(xí)使用示例
這篇文章主要為大家介紹了Python的getattr方法學(xué)習(xí)使用示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-08-08
python線程池 ThreadPoolExecutor 的用法示例
這篇文章主要介紹了python線程池 ThreadPoolExecutor 的用法示例,幫助大家更好得理解和使用python線程池,感興趣的朋友可以了解下2020-10-10
python讀寫(xiě)csv文件方法詳細(xì)總結(jié)
在本文中小編給各位分享的是關(guān)于python讀寫(xiě)csv文件方法的詳細(xì)內(nèi)容,對(duì)此有需要的朋友們跟著學(xué)習(xí)參考下。2019-07-07
用python實(shí)現(xiàn)五子棋實(shí)例
這篇文章主要為大家詳細(xì)介紹了用python實(shí)現(xiàn)五子棋實(shí)例,文中示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2022-05-05
詳解pycharm自動(dòng)import所需的庫(kù)的操作方法
這篇文章主要介紹了pycharm自動(dòng)import所需的庫(kù)的操作方法,本文通過(guò)圖文并茂的形式給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-11-11
詳解用Python處理HTML轉(zhuǎn)義字符的5種方式
本文介紹了詳解用Python處理HTML轉(zhuǎn)義字符的5種方式,小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2017-12-12

