Tensorflow - Изменить псевдоним тензора SavedModel - PullRequest
Новая
       17

Tensorflow - Изменить псевдоним тензора SavedModel

0 голосов
/ 06 ноября 2019

Я хочу использовать свою модель в Google ML, и мои данные представляют собой большое изображение. Я прочитал в документации, что для того, чтобы это работало, мне нужно изменить свой входной тензор, чтобы он заканчивался на _bytes.

Я попытался загрузить SavedModel и сохранить его снова, но некоторые из выходных тензоров не существуют вграфик, есть ли способ переименовать тензор внутри модели? 

import tensorflow as tf
from tensorflow.python.saved_model import signature_constants
from tensorflow.python.saved_model import tag_constants


builder = tf.saved_model.builder.SavedModelBuilder("path")
sigs = {}

with tf.Session(graph=tf.Graph()) as sess:
    tf.saved_model.loader.load(sess, ["serve"], "path")
    g = tf.get_default_graph()
    sigs[signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY] = \
            tf.saved_model.signature_def_utils.predict_signature_def(
                {"image_tensor_bytes": image_tensor },
                {output_tensors})

    builder.add_meta_graph_and_variables(sess, [tag_constants.SERVING], signature_def_map=sigs)

builder.save()
...