提交 d46c8e83 编写于 作者: Q qiaolongfei

add test to graph

上级 d91cfe66
......@@ -41,5 +41,9 @@ def load_model(model_pb_path):
return json.dumps(json_obj, sort_keys=True, indent=4, separators=(',', ': '))
json_str = load_model("/Users/qiaolongfei/project/onnx/model/inception_v1/model.pb")
print(json_str)
if __name__ == '__main__':
import os
import sys
current_path = os.path.abspath(os.path.dirname(sys.argv[0]))
json_str = load_model(current_path + "/mock/inception_v1.pb")
print(json_str)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册