【发布时间】:2020-02-25 23:36:57
【问题描述】:
我是 tensorflow 的新手,我有一个这种格式的 numpy 输入数据:
x_train_n.shape = (200,64,2048)
意思是200是训练数据集的个数,64是H,2048是W
当我想将此输入提供给我的网络时,首先我必须对其进行重塑:
x_train_n = x_train_n.reshape(x_train_n.shape[0], 1, rows, cols)
然后
inputs = Input(shape=x_train_n.shape[1:])
output1 = Conv2D(32, (3, 15), strides=(1, 2), padding='same', data_format='channels_first', input_shape=x_train_n.shape[1:])(inputs)
否则我得到 Conv2d 预期有 4 个维度但维度为 3 个错误的错误。
这是正确的做法吗?如果是这样,为什么这有意义?
为什么不重塑就不能跟随?
output1 = Conv2D(32, (3, 15), strides=(1, 2), padding='same', data_format='channels_first', input_shape=x_train_n.shape())(inputs)
【问题讨论】:
标签: python tensorflow input keras deep-learning