python - 来自 numpy 数组的 Keras 自定义数据生成器

Question

我有两个包含 image and label 的 numpy 变量数据分别。有500 labeled image ，形状每image is 240 x 240.

import numpy as np
images = np.random.randint(4, size=(500,240,240))
labels =  np.random.rand(500,240,240)

如何使用 Keras 生成器进行模型训练？ 在此先感谢您的帮助。

Answer 1

如果您愿意对图像进行一些小的更改，则可以轻松完成此操作。基本上你需要在 images 上再添加一个维度( channel 维度)。

import numpy as np
import tensorflow as tf

images = np.expand_dims(np.random.randint(4, size=(500,240,240)),-1)
labels =  np.random.rand(500,240,240)

gen = tf.keras.preprocessing.image.ImageDataGenerator()
res = gen.flow(images, labels)
x, y = next(res)

您可以通过创建另一个生成器生成 Keras 生成器的数据并删除该维度来后期处理和删除此维度。