是否有任何库可为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。