[英]Tensorflow : load saved model : UnicodeDecodeError
我有一个由另一个人创建的Tensorflow模型,我想从中创建一个冻结图。
我有的文件:
+--- name.txt
+--- saved_model.pb
+--- variables
| +--- variables.data-00000-of-00001
| +--- variables.index
因此,借助此页面 ,我做了以下代码:
import tensorflow as tf
import os
pwd = os.getcwd()
toolsFilepath = os.path.dirname(tf.__file__)+"/python/tools"
modelFolder = [PATH_TO_MODEL]
networkArcFile = "saved_model.pb"
networkWeightsFile = "variables/variables.data-00000-of-00001"
outputFolder = pwd+"/output/"+networkArcFile+".modelzoo"
args = "--input_graph="+modelFolder+"/"+networkArcFile+"\
--input_checkpoint="+modelFolder+"/"+networkWeightsFile+"\
--output_graph="+outputFolder
os.system(toolsFilepath+"/freeze_graph.py "+args)
但是当我这样做时,我得到了错误:
Traceback (most recent call last):
File "[MY_PATH]\lib\site-packages\tensorflow\python\tools\freeze_graph.py", line 491, in <module>
run_main()
File "[MY_PATH]\lib\site-packages\tensorflow\python\tools\freeze_graph.py", line 488, in run_main
app.run(main=my_main, argv=[sys.argv[0]] + unparsed)
File "[MY_PATH]\lib\site-packages\tensorflow\python\platform\app.py", line 40, in run
_run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)
File "[MY_PATH]\lib\site-packages\absl\app.py", line 300, in run
_run_main(main, args)
File "[MY_PATH]\lib\site-packages\absl\app.py", line 251, in _run_main
sys.exit(main(argv))
File "[MY_PATH]\lib\site-packages\tensorflow\python\tools\freeze_graph.py", line 487, in <lambda>
my_main = lambda unused_args: main(unused_args, flags)
File "[MY_PATH]\lib\site-packages\tensorflow\python\tools\freeze_graph.py", line 381, in main
flags.saved_model_tags, checkpoint_version)
File "[MY_PATH]\lib\site-packages\tensorflow\python\tools\freeze_graph.py", line 340, in freeze_graph
input_graph_def = _parse_input_graph_proto(input_graph, input_binary)
File "[MY_PATH]\lib\site-packages\tensorflow\python\tools\freeze_graph.py", line 253, in _parse_input_graph_proto
text_format.Merge(f.read(), input_graph_def)
File "[MY_PATH]\lib\site-packages\tensorflow\python\lib\io\file_io.py", line 132, in read
pywrap_tensorflow.ReadFromStream(self._read_buf, length, status))
File "[MY_PATH]\lib\site-packages\tensorflow\python\lib\io\file_io.py", line 100, in _prepare_value
return compat.as_str_any(val)
File "[MY_PATH]\lib\site-packages\tensorflow\python\util\compat.py", line 107, in as_str_any
return as_str(value)
File "[MY_PATH]\lib\site-packages\tensorflow\python\util\compat.py", line 80, in as_text
return bytes_or_text.decode(encoding)
UnicodeDecodeError: 'utf-8' codec can't decode byte 0x83 in position 3: invalid start byte
该错误的可能原因是什么,我该如何解决?
注意事项:
--input_binary=true
处为类似的UnicodeDecodeError所建议的freeze_graph.py
,尝试在调用freeze_graph.py
时尝试使用--input_binary=true
:'utf8'编解码器无法解码字节...问题。 您的saved_model.pb
最有可能是二进制文件。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.