无法在TF 1.0.1中将freeze_graph.py与tf.contrib.image.rotate（）一起使用

Question

我目前使用tf.contrib.image.rotate()来增加我的训练数据。当我尝试冻结相应的图表时，我收到以下错误：

  

ValueError：在定义的操作中没有名为ImageProjectiveTransform的操作。

当我从图表中删除tf.contrib.image.rotate()时，freeze_graph.py正常结束。

为什么会出现此错误？什么是最简单的解决方法？

Answer 1

发生此错误是因为freeze_graph.py无法访问contrib操作系统（例如您正在使用的操作系统）。我认为您需要提交github问题才能获得永久性修复，但临时修复是将tensorflow.contrib.image的导入添加到freeze_graph.py并重新运行。