python - keras zca_whitening - 没有错误，没有生成输出

Question

使用zca_whitening时，我的代码卡在某个地方，既不显示任何错误，也不显示输出。当我跳过 zca_whitening 并应用其他转换时，代码运行完美。我在这里附上代码片段。如果我在这里做错了什么，请帮助我:

datagen = ImageDataGenerator(zca_whitening=True)
datagen.fit(x_train)

其中 >> x_train 是训练图像集 (dim = 50 x 64 x 64 x 3) 。运行 datagen.fit 后，代码没有显示进一步的输出或错误，似乎进程运行了无限的时间段。

另一方面，这些转换工作得很好:

datagen = ImageDataGenerator(rotation_range=0.90,
    width_shift_range=0.2,
    height_shift_range=0.2,
    fill_mode='nearest',
    horizontal_flip=True,
    vertical_flip=True)
datagen.fit(x_train)

我在这里缺少什么吗？

Answer 1

将 x_train 修改为形状 (3, 64, 64)。

您可以使用以下代码来做到这一点:x_train = x_train.transpose((2,1,0))

这主要是由于theano和tensorflow后端之间的切换造成的。检查 Keras 文档中的 dim_order。