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

Question

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

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

这里>>x_train是训练图像集（暗= 50×64×64×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。