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

Question

我目前在本地笔记本电脑中有图像，并且想要将其转换为数据集以便加载keras python。我有什么解决办法吗？

Answer 1

您可以使用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')

https://keras.io/preprocessing/image/

Answer 2

在此link中，您可以找到几种使用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]))

不确定它是否最有效。