如何将图像文件数据集加载到TensorFlow Jupyter笔记本

import tensorflow as tf
import os
import glob

dataset_path = '/path/to/dataset'

image_paths = glob.glob(os.path.join(dataset_path, '*.jpg'))

def preprocess_image(image_path):
    image = tf.io.read_file(image_path)
    image = tf.image.decode_jpeg(image, channels=3)
    image = tf.image.resize(image, [224, 224])
    image = image / 255.0  # 归一化到 [0, 1] 范围
    return image

dataset = tf.data.Dataset.from_tensor_slices(image_paths)

dataset = dataset.map(preprocess_image)

labels_path = '/path/to/labels.txt'

def load_labels(labels_path):
    with open(labels_path, 'r') as f:
        labels = f.read().splitlines()
    return labels

labels = load_labels(labels_path)
dataset = dataset.map(lambda x: (x, labels))

dataset = dataset.shuffle(buffer_size=len(image_paths))
dataset = dataset.batch(batch_size)

for image, label in dataset:
    # 在这里执行你的训练代码
    pass