张量流中不正确的尺寸进入卷积层

时间:2020-01-17 04:25:19

标签: python tensorflow keras

我有15张尺寸为28x28的图像。因此,我的images.shape是(15,28,28)。但是,出现此错误:

ValueError: Error when checking input: expected conv2d_22_input to have 4 dimensions, but got array with shape (15, 28, 28)

我尝试了各种重塑技术,并且可以将其用于一个形状为(1、28、28、1)的图像。我要如何处理图像列表?

2 个答案:

答案 0 :(得分:2)

您需要将渠道尺寸重新添加到批次中。根据您的配置将其重塑为(15,28,28,1)(15,1,28,28)(默认为频道在keras中居最后)。

答案 1 :(得分:0)

numpy.expand_dims(images, axis=4)为我工作