加载Tensorflow模型时出现“ImageProjectiveTransform”键错误

2024-03-28 11:45:41 发布

您现在位置:Python中文网/ 问答频道 /正文

我在MNIST上训练了一个ConvNet,并使用simple\u save保存了模型。你知道吗

加载保存的模型时,使用tf.saved_model.loader.load(sess,[tag_constants.SERVING],'/save_folder')

我得到这个错误KeyError: u'ImageProjectiveTransform

保存ConvNet模型的代码

graph = tf.get_default_graph()

from tensorflow.python.saved_model import tag_constants
with graph.as_default():
  input_dict = {"x":x,"y":y,"keep_prob":keep_prob,"aug_img":aug_img}
  output_dict = {"logits":fc3}
  tf.saved_model.simple_save(sess,'/sample_data/',input_dict,output_dict)

我使用tf.contrib.image.rotate作为扩充技术,但在保存模型时没有将其添加到任何dict中。你知道吗

错误是由此产生的吗?还是有其他问题?你知道吗


Tags: 模型defaultinputmodelsavetftag错误