2017-04-07 81 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

相關問題