2017-04-07 85 views
0

我目前使用tf.contrib.image.rotate()来增强我的训练数据。当我尝试冻结对应的图,收到以下错误:无法在TF 1.0.1中使用带tf.contrib.image.rotate()的freeze_graph.py

ValueError: No op named ImageProjectiveTransform in defined operations.

当我从图中删除tf.contrib.image.rotate()freeze_graph.py饰面正常。

为什么会发生此错误?什么是最简单的工作?

回答

1

发生此错误是因为freeze_graph.py无权访问contrib操作(就像您正在使用的那样)。我认为您需要提交github问题才能获得永久性修复,但临时修复方法是将tensorflow.contrib.image添加到freeze_graph.py并重新运行。

+0

您的解决方案。谢谢!我现在的问题是,为什么我不需要对我在图中使用的'tf.contrib.slim'使用相同的解决方法? – RobR

+0

苗条不添加任何新的操作内核,只是一个更好的用户界面使用它们 –

+0

有道理。谢谢。 – RobR

相关问题