第一種方式毡证,使用TensorFlow自帶的tensorflow\python\tools\import_pb_to_tensorboard.py腳本,先把凍結(jié)圖模型轉(zhuǎn)換為TensorBoard可讀取的格式痛悯,然后啟動TensorBoard查看模型結(jié)構(gòu)歹河,包括輸入和輸出
范例:
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=.\
第二種方式,使用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