Как иметь несколько выходов в тензорном потоке SavedModel? - PullRequest
0 голосов
/ 03 мая 2020

Таким образом, вся история в том, что я пытаюсь преобразовать замороженный граф логического вывода .pb в модель tflite, и для этого я сначала пытаюсь создать SavedModel. Вот код, который я пытаюсь использовать ниже:

with tf.Session(graph=tf.Graph()) as sess:
    # name="" is important to ensure we don't get spurious prefixing
    tf.import_graph_def(graph_def, name="")
    g = tf.get_default_graph()
    inp = g.get_tensor_by_name("image_tensor:0")

    out = {{g.get_tensor_by_name('num_detections:0')}, {g.get_tensor_by_name('detection_boxes:0')},{g.get_tensor_by_name('detection_scores:0')},{g.get_tensor_by_name('detection_classes:0')}}


    sigs[signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY] = \
        tf.saved_model.signature_def_utils.predict_signature_def(
            {"inputs": inp}, {"outputs": out})

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

builder.save()

Я неправильно выполняю 'out', однако я не знаю, как включить больше выходов для сигнатуры SavedModel, или если это так даже возможно?

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