Tensorflow 如何将图像作为keras python的数据集

我目前在本地笔记本电脑中有图像，希望将它们转换为数据集，以便加载到keras python中。有什么解决方案吗？

您可以使用

ImageDataGenerator

加载图像

        train_datagen = ImageDataGenerator(
        rescale=1./255,
        shear_range=0.2,
        zoom_range=0.2,
        horizontal_flip=True)

        train_generator = train_datagen.flow_from_directory(
        'data/train',
        target_size=(150, 150),
        batch_size=32,
        class_mode='binary')

在这里，您可以找到几种使用Python从目录加载图像的方法。以下是一个简单的方法：

import numpy as np
import cv2
import os

instances = []

# Load in the images
for filepath in os.listdir('images/'):
   instances.append(cv2.imread('images/{0}'.format(filepath),0))

print(type(instances[0]))

但不确定这是否是最有效的