我执行此命令来检查张量流训练的结果。
python label_image.py --image xxx.jpg --graph retrained_graph.pb --labels retrained_labels.txt
并发现以下错误。
/Users/xxx/anaconda3/lib/python3.6/site-packages/h5py/__init__.py:36: FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.
from ._conv import register_converters as _register_converters
2018-07-11 00:39:22.028051: I tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
Traceback (most recent call last):
File "label_image.py", line 131, in
input_operation = graph.get_operation_by_name(input_name)
File "/Users/xxx/tensorFlow/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 3718, in get_operation_by_name
return self.as_graph_element(name, allow_tensor=False, allow_operation=True)
File "/Users/xxx/tensorFlow/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 3590, in as_graph_element
return self._as_graph_element_locked(obj, allow_tensor, allow_operation)
File "/Users/xxx/tensorFlow/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 3650, in _as_graph_element_locked
"graph." % repr(name))
KeyError: "The name 'import/Mul' refers to an Operation not in the graph."
label_image.py是以下几行。
input_height = 299
input_width = 299
input_mean = 0
input_std = 255
input_layer = "Mul"
output_layer = "final_result"
答案 0 :(得分：0)
python label_image.py --image cat3.jpg --graph retrained_graph.pb --labels retrained_labels.txt --input_layer =占位符
此命令，问题已解决。