【发布时间】:2021-12-16 08:29:27
【问题描述】:
我无法将适当的输入形状传递给具有 Conv2D 层的基于 CNN 的网络。 最初,这些是我的火车形状。我的火车数据被重新塑造成窗口:
X_train: (7,100,5185)= (number of features, window size, number of windows)
y_train= (5185, 100 ) = one labeled column that is also windowed
然后我根据这些数据计算一些递归图,然后我将得到这些形状:
X_train_rp= (5185, 100,100, 7), 100 * 100 referring to my images
y_train = (5185, 100 ), remains unchanged
我将这两个传递给基于 conv2D 的 CNN:
model.add(layers.Conv2D(64, kernel_size=3, activation='relu', input_shape=(100, 100, 7)))
我收到此错误: Data cardinality is ambiguous: x sizes: 100, 100, 100 ......... y sizes: 5185 Make sure all arrays contain the same number of samples.
我尝试了许多形状组合,但都没有成功!我做错了什么??
编辑: 这是使用的模型定义
import tensorflow as tf
X_train_rp = tf.zeros((10, 100,100, 7))
y_train = tf.zeros((10, 100))
#create model
model = tf.keras.Sequential() #add model layers
model.add(tf.keras.layers.Conv2D(64, kernel_size=3, activation='relu',
data_format='channels_last', input_shape=(100, 100, 7)))
model.add(tf.keras.layers.Conv2D(32, kernel_size=3, activation='relu'))
model.add(tf.keras.layers.Flatten())
model.add(tf.keras.layers.Dense(2, activation='softmax'))
#compile model using accuracy to measure model performance
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
model.fit(X_train_rp, y_train_shaped, epochs=3)
model.predict(X_train_rp)
【问题讨论】:
-
如果问题只是关于形状,前两个代码块可以从问题中排除,因为只有
X_train_rp和y_train的实际输入形状是相关的。另一方面,必须包含最相关的部分,即您的模型定义。
标签: python image conv-neural-network shapes training-data