python - 将 Keras 增强数据保存为 numpy 数组

Question

使用kerasImageDataGenerator，我们可以将增强图像保存为 png 或 jpg :

    for X_batch, y_batch in datagen.flow(train_data, train_labels, batch_size=batch_size,\
                save_to_dir='images', save_prefix='aug', save_format='png'):

我有一个形状为 (1600, 4, 100,100) 的数据集，这意味着 1600 张图像，4 个 100x100 像素 channel 。如何将增强数据保存为形状 (N,4,100,100) 的 numpy 数组而不是单个图像？

Answer 1

由于您知道样本数 = 1600，因此只要达到此数字，您就可以停止 datagen.flow()。

augmented_data = []
num_augmented = 0
for X_batch, y_batch in datagen.flow(train_data, train_labels, batch_size=batch_size, shuffle=False):
    augmented_data.append(X_batch)
    num_augmented += batch_size
    if num_augmented == train_data.shape[0]:
        break
augmented_data = np.concatenate(augmented_data)
np.save(...)

请注意，您应该正确设置batch_size(例如batch_size=10)，以便不会生成额外的增强图像。