TensorFlow批读取图片

本文介绍如何使用TensorFlow读取指定路径下的所有JPEG图像文件,并将这些图像转换为tf.float32数据类型,以便于进行深度学习模型训练。代码示例展示了从文件读取到图像解码及数据类型转换的完整过程。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

# coding:utf-8

from __future__ import print_function
import os
import tensorflow as tf


def TF_image_data(path_to_images):  
    """
    using tensorflow to read and store data
    """
    if not tf.gfile.Exists(path_to_images):
        print("path to images does not exist")
    else:
        #search all jpeg files
        path_img_list = files = os.listdir(path_to_images)
        #path_img_list = tf.gfile.Glob(os.path.join(path_to_images, '*.jpg'))
        print(path_img_list)
        total_img = []
        for file in path_img_list:
            print("the name of jpeg file:", file)
            img_jpg = tf.gfile.FastGFile(path_to_images +'/' +file,'rb').read() 
            img_decode = tf.image.decode_jpeg(img_jpg)
            img_data = tf.image.convert_image_dtype(img_decode, dtype = tf.float32)
            total_img.append(img_data)
        print(len(total_img))
        return total_img 


if __name__ == '__main__':
    path_to_images = './Images'
    with tf.Session() as sess:
        TF_image_data(path_to_images)

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值