我是卷积神经网络的新手。我没有以图像格式获取数据,而是得到了扁平图像矩阵,即[10000x784]。
表示 10000 张尺寸为 28x28 的图像
考虑到一张图像大小为28x28,我应该如何将数据矩阵提供给 CNN 的输入?
我的模型是:
model = models.Sequential()
model.add(layers.Conv2D(64, (3, 3), activation='relu', input_shape=(28,28,1)))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Conv2D(64, (3, 3), activation='relu'))
#model.add(layers.Flatten())
model.add(layers.Dense(2500, activation='relu'))
model.add(layers.Dense(2500, activation='relu'))
model.add(layers.Dense(1, activation='relu'))
model.compile(optimizer='adam',
loss='mean_squared_error',
metrics=['mae','mse'])
callback = tf.keras.callbacks.EarlyStopping(monitor='val_loss', patience=15)
#Fits model
history= model.fit(x_trained, y_train, epochs = 7000, validation_split = 0.2, shuffle= True, verbose = 1, callbacks=[callback])
我在 model.fit 处遇到错误。
P.S:我正在做回归,对于每张图像我都有一个值作为输出
最佳答案
从Reshape
层开始:
model = models.Sequential()
model.add(layers.Reshape((28, 28, 1), input_shape=(784,)))
model.add(layers.Conv2D(64, (3, 3), activation='relu'))
# ...
关于python - 我有 10000 个矢量形式的图像,如何将其转换为我的卷积神经网络?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/58918032/