tensorflow.python.framework.errors_impl.NotFoundError:FeedInputs:无法找到提要输出conv2d_input_7:0

时间:2019-01-04 21:12:48

标签: python tensorflow keras

由于硬件要求,我正在使用张量流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')

我在做什么错了?

0 个答案:

没有答案