tensorflow如何批量读取图片
作者:雪儿waii 时间:2023-10-25 03:50:25
本文实例为大家分享了tensorflow如何批量读取图片的具体代码,供大家参考,具体内容如下
代码:
import tensorflow as tf
import os
def picread(filelist):
"""
读取狗的图片并转换成张量
:param filelist: 文件路f径+名字的列表
:return: 每张图片的张量
"""
# 1.构造文件的队列
file_queue = tf.train.string_input_producer(filelist)
# 2.构造阅读器去读取图片内容(默认读取一张图片)
reader = tf.WholeFileReader()
key,value = reader.read(file_queue)
# 3.对读取的图片进行解码
image = tf.image.decode_jpeg(value)
# 4.处理图片的大小(统一大小)
image_resize = tf.image.resize_images(image,[200,200])
# 注意:一定要把样本的形状固定,在批处理中要求所有数据的形状必须固定
image_resize.set_shape([200,200,3])
# 5.进行批处理
image_resize_batch = tf.train.batch([image_resize],batch_size=3,num_threads=1,capacity=3)
return image_resize
#批处理大小,跟队列,数据的数量没有影响,只决定 这批次处理多少数据
if __name__ == "__main__":
# 1.找到文件,放入列表 路径+名字 ->列表当中
file_name = os.listdir("./data/dogpic/")
filelist = [os.path.join("./data/dogpic/",file) for file in file_name ]
image_batch= picread(filelist)
#开启会话运行结果
with tf.Session() as sess:
#定义一个线程协调器
coord = tf.train.Coordinator()
#开启读文件的线程
threads = tf.train.start_queue_runners(sess,coord=coord)
#打印读取的内容
print(sess.run([image_batch]))
#回收子线程
coord.request_stop()
coord.join(threads)
结果:
来源:https://blog.csdn.net/XUEER88888888888888/article/details/86666614
标签:tensorflow,批量读取,读取图片
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
Pycharm激活码激活两种快速方式(附最新激活码和插件)
2023-05-17 12:07:07
Python数据清洗工具之Numpy的基本操作
2023-08-31 02:39:10
![](https://img.aspxhome.com/file/2023/0/66300_0s.png)
MySQL新手入门指南--快速参考
2024-01-21 09:58:14
ASP.NET MVC把数据库中枚举项的数字转换成文字
2024-01-16 07:35:39
Python中使用json.load()和json.loads()加载json数据的方法实例
2021-12-15 20:12:31
BERT vs GPT自然语言处理中的关键差异详解
2022-04-01 08:15:36
Python代码块及缓存机制原理详解
2023-07-02 08:12:07
轻松读懂Golang中的数组和切片
2024-02-07 16:25:48
微信小程序封装多张图片上传api代码实例
2024-04-25 13:13:31
用python与文件进行交互的方法
2021-05-22 12:39:35
python使用Matplotlib绘制多种常见图形
2023-05-26 05:06:28
![](https://img.aspxhome.com/file/2023/4/101594_0s.webp)
Python实现深度遍历和广度遍历的方法
2023-10-13 20:41:13
10个有用的Python字符串函数小结
2023-08-13 11:26:28
Python reshape的用法及多个二维数组合并为三维数组的实例
2021-12-18 10:29:25
![](https://img.aspxhome.com/file/2023/4/102324_0s.jpg)
Go语言单链表实现方法
2024-02-14 15:46:45
一个PHP的QRcode类与大家分享
2023-06-24 05:39:33
![](https://img.aspxhome.com/file/2023/5/55315_0s.gif)
Python实现对PPT文件进行截图操作的方法
2021-10-14 13:25:19
在Linux系统上通过uWSGI配置Nginx+Python环境的教程
2023-01-05 22:25:57
微信小程序实现获取自己所处位置的经纬度坐标功能示例
2024-04-10 13:57:59
![](https://img.aspxhome.com/file/2023/0/136870_0s.png)
老生常谈Python进阶之装饰器
2022-05-20 10:51:23