且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

使用ImageDataGenerator(Keras)时,fit_generator输入尺寸错误

更新时间:2023-12-01 23:47:04

已解决-应该应该是:

autoencoder = Sequential()
autoencoder.add(Reshape((784,), input_shape=(28,28,1)))
autoencoder.add(Dense(30, activation='relu'))
autoencoder.add(Dense(784, activation='relu'))
.
.
.

autoencoder.fit_generator(
        datagen.flow(x_train, x_train.reshape(len(x_train),784,), batch_size=8),
        steps_per_epoch=int(len(x_train)/8),
        epochs=64,
       )