Могу ли я получить модель Keras из файла .pb? - PullRequest
0 голосов
/ 08 ноября 2019

У меня есть файл 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')

Я хочу сгенерировать керас как файл резюме

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...