查看keras模型的输入输出节点名称
有两种方式
1.直接调用model.inputs和model.outputs
2.利用.pb文件查看,代码如下
import tensorflow as tf
import os
model_dir = './'
model_name = 'frozen_graph.pb'
def create_graph():
with tf
查看keras模型的输入输出节点名称
有两种方式
1.直接调用model.inputs和model.outputs
2.利用.pb文件查看,代码如下
import tensorflow as tf
import os
model_dir = './'
model_name = 'frozen_graph.pb'
def create_graph():
with tf