查看TensorFlow冻结图模型构架的两种方式
2021-06-14 本文已影响0人
LabVIEW_Python
第一种方式，使用TensorFlow自带的tensorflow\python\tools\import_pb_to_tensorboard.py脚本，先把冻结图模型转换为TensorBoard可读取的格式，然后启动TensorBoard查看模型结构，包括输入和输出
范例：
python C:\Users\Administrator\anaconda3\envs\tf_gpu\Lib\site-packages\tensorflow\python\tools\import_pb_to_tensorboard.py --model_dir ssd_mobilenet_v1_coco_2018_01_28 --log_dir ./
tensorboard --logdir=.\
运行import_pb_to_tensorboard.py
用TensorBoard查看模型结构
第二种方式，使用OpenVINO自带\deployment_tools\model_optimizer\mo\utils、summarize_graph.py脚本，使用范例如下
python summarize_graph.py --input_model d:\training\ssd_mobilenet_v1_coco_2018_01_28\frozen_inference_graph.pb
运行summarize_graph.py
