python - keras中卷积神经网络的输入形状

Question

我正在尝试使用 cnn 构建图像分类器。我的图像大小为 (256,256) 像素。

如果我通过将输入形状设置为 (64,64) 或 (128,128) 来训练 cnn 会发生什么，因为 (256,256) 将花费大量时间来处理？

标签： pythonkerasclassificationconv-neural-network

Answer 1

它会抛出一个错误。您可以使用 cv2.resize() 调整图像大小，也可以将正确的输入形状放在 cnn 层中，然后放置一个 maxpooling 层以减少参数数量。