machine-learning - 是否有任何库可以为 keras.Model.fit_generator 构建生成器？

Question

keras.preprocessing.image.ImageDataGenerator用于为图像输入创建生成器。

是否有任何类似的库可用于正常的 numpy 数组输入？

Answer 1

不，我没有遇到任何这样的库，但我们可以编写一个生成器来为 numpy 数组执行此操作。

def gen(batch):
     j = 0 
     while True:
        x = np.array(x_shape)
        y = np.array(y_shape)
        for i in range(batch_size):
              x = read(batch[j])
              y = read(batch[j])
        j += 1
        yeild x,y

我们可以将其用作模板，我们正在使用生成器，因为我们的数据不适合 RAM。当使用多处理时，我们也可以使用迭代器而不是 j。