tensorflow CIFAR 100数据(50000,32,32,3)形状改变为(50000,32,32)形状

yqlxgs2m  于 2023-01-31  发布在  其他
关注(0)|答案(1)|浏览(148)

我已经从tf.keras.datasets下载了cifar100数据,如何将数据(50000,32,32,3)形状更改为(50000,32,32)以进行训练?
不改变形状也能训练吗?

eagi6jfj

eagi6jfj1#

您可以在不将数据形状从(50000,32,32,3)更改为(50000,32,32)的情况下定型模型。在定义定型模型时,您需要使用RGB channel = 3表示input_shape,如下所示:

model = models.Sequential()
model.add(layers.Conv2D(32, (3, 3), activation='relu', input_shape=(32, 32, 3)))

有关使用cifar10数据集进行模型训练的更多详细信息,请查看此link

相关问题