由于硬件要求,我正在使用张量流1.9.0。 我已经在Google colab上训练了一个基本的CNN网络,并尝试在本地计算机上使用anaconda spyder来使用保存的训练模型进行预测。
收到以下错误: tensorflow.python.framework.errors_impl.NotFoundError:FeedInputs:无法找到提要输出conv2d_input_7:0
我用来定义模型的代码:
const
然后加载和使用模型,因为网络用于图像的不同部分,所以我有一个分别在每个帧上运行的for循环。
import tensorflow as tf
from tensorflow import keras
#Build the model
model = keras.Sequential([
keras.layers.Conv2D(filters=32, kernel_size=(3,3), strides=(2, 2), activation=tf.nn.relu,input_shape=(64, 64,3)),
keras.layers.MaxPooling2D(pool_size=(2,2)),
keras.layers.Flatten(),
keras.layers.Dense(128, activation=tf.nn.relu),
keras.layers.Dense(6, activation=tf.nn.softmax),
])
#Compile the model
model.compile(optimizer=tf.keras.optimizers.Adam(lr=0.001),
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
# Train the model
model.fit(train_images, train_labels,batch_size=64, epochs=10)
#save the model
model.save(saved_model_path+'my_model.h5')
我在做什么错了?