tensorflow批量读取图片

最新推荐文章于 2021-11-23 14:24:52 发布

雪儿waii

最新推荐文章于 2021-11-23 14:24:52 发布

阅读量6.3k

点赞数 2

分类专栏： tensonflow

本文链接：https://blog.csdn.net/XUEER88888888888888/article/details/86666614

版权

本文介绍如何利用TensorFlow库高效地批量读取和预处理图像数据，为深度学习模型训练做好准备。通过示例代码，展示了如何组织数据集，并进行批处理操作。

摘要由CSDN通过智能技术生成

在这里插入图片描述

代码：

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