У меня есть файл ptocolbuffer для модели VGG. Мне нужно сгенерировать модель keras. Могу ли я использовать файл .pb для генерации модели Keras, чтобы я мог использовать такие функции Keras, как сводка?
Я загрузил файл .pb в tf и сгенерировал софиги моделей и веса слоев.
import tensorflow as tf
from tensorflow.python.platform import gfile
from tensorflow.python.framework import tensor_util
GRAPH_PB_PATH = './small_vgg_tf_graph.pb'
with tf.Session() as sess:
print("load graph")
with gfile.FastGFile(GRAPH_PB_PATH,'rb') as f:
graph_def = tf.GraphDef()
graph_def.ParseFromString(f.read())
sess.graph.as_default()
tf.import_graph_def(graph_def, name='')
graph_nodes=[n for n in graph_def.node]
names = []
for t in graph_nodes:
names.append(t.name)
with open('names.txt','w') as fh:
for i in names:
fh.write(str(i)+'\n')
weight_nodes = [n for n in graph_def.node if n.op == 'Const']
with open('weights.txt','w') as fh:
for n in weight_nodes:
fh.write("Name of the node - %s" % n.name+'\n')
fh.write("Value - " )
fh.write(str(tensor_util.MakeNdarray(n.attr['value'].tensor))+'\n')
with open('ops.txt','w') as fh:
for op in sess.graph.get_operations():
fh.write(str(op)+'\n')
Я хочу сгенерировать керас как файл резюме