tf.train.import_meta_graph 报错
Traceback (most recent call last):
File "freeze_graph.py", line 109, in <module>
main(parse_arguments(sys.argv[1:]))
File "freeze_graph.py", line 50, in main
saver = tf.train.import_meta_graph(os.path.join(model_dir_exp, meta_file), clear_devices=True)
File "/root/anaconda3/envs/tsp/lib/python3.6/site-packages/tensorflow/python/training/saver.py", line 1909, in import_meta_graph
**kwargs)
File "/root/anaconda3/envs/tsp/lib/python3.6/site-packages/tensorflow/python/framework/meta_graph.py", line 737, in import_scoped_meta_graph
producer_op_list=producer_op_list)
File "/root/anaconda3/envs/tsp/lib/python3.6/site-packages/tensorflow/python/util/deprecation.py", line 432, in new_func
return func(*args, **kwargs)
File "/root/anaconda3/envs/tsp/lib/python3.6/site-packages/tensorflow/python/framework/importer.py", line 431, in import_graph_def
_RemoveDefaultAttrs(op_dict, producer_op_list, graph_def)
File "/root/anaconda3/envs/tsp/lib/python3.6/site-packages/tensorflow/python/framework/importer.py", line 211, in _RemoveDefaultAttrs
op_def = op_dict[node.op]
KeyError: 'ImageProjectiveTransform'
解决办法:
import tensorflow.contrib.image
参考: