加载Tensorflow模型时出现KeyError u'ImageProjectiveTransform'

Question

我在 MNIST 上训练了一个ConvNet，并使用 simple_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作为扩充技术，但在保存模型时并未将其添加到任何字典中。

是由于该错误引起的吗？还是还有其他问题？