gpt4 book ai didi

python - 我有 10000 个矢量形式的图像,如何将其转换为我的卷积神经网络?

转载 作者:行者123 更新时间:2023-12-01 06:53:07 25 4
gpt4 key购买 nike

我是卷积神经网络的新手。我没有以图像格式获取数据,而是得到了扁平图像矩阵,即[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/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com