Как изменить значение тензора в замороженном файле pb для Tensorflow? - PullRequest
Новая
       16

Как изменить значение тензора в замороженном файле pb для Tensorflow?

0 голосов
/ 29 января 2019

Теперь у меня есть замороженный файл pb tenorflow.И я изменяю некоторые значения тензора, полученного из указанного имени слоя conv1/weights: 

 import tensorflow as tf 
 from tensorflow.core.framework import graph_pb2
 from tensorflow.python.framework import tensor_util
 FLAGS = tf.app.flags.FLAGS
 tf.app.flags.DEFINE_string('input_frozen_graph_path',
                           '/tmp/model.pb', '')
 graph = tf.GraphDef()
 with tf.gfile.Open(FLAGS.input_frozen_graph_path, 'rb') as f:
     data = f.read()
     graph.ParseFromString(data)

 nodes = {node.name: index for index, node in enumerate(graph.node)}
 conv1_weights = tensor_util.MakeNdarray(graph.node[nodes["conv1/weights"]].attr['value'].tensor)
 #modify some values in conv1_weights

, но как записать это измененное значение conv1_weights обратно в узел с именем conv1/weights для графа.узел?Кто-нибудь может дать несколько советов?спасибо

...