Python numpy阵列中的Keras自定义数据生成器

我有两个numpy变量，分别包含

image和label

数据。有

500个标记图像

，每个

图像的形状为240 x 240。

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

---

如何为模型培训制作Keras生成器？提前感谢您的帮助。

如果您愿意对图像做一些小改动，您可以轻松完成此操作。基本上，您需要向

图像

（频道维度）再添加一个维度

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

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)