tensorflow - Unable to use freeze_graph.py with tf.contrib.image.rotate() in TF 1.0.1 -
i use tf.contrib.image.rotate() augment training data. when try freeze corresponding graph receive following error:
valueerror: no op named imageprojectivetransform in defined operations.
when remove tf.contrib.image.rotate() graph, freeze_graph.py finishes normally.
why error occur? easiest work around?
this error happens because freeze_graph.py doesn't have access contrib ops (like 1 you're using). think need file github issue permanent fix, temporary fix add import of tensorflow.contrib.image freeze_graph.py , rerun.
Comments
Post a Comment