Как использовать замороженный граф вывода для обнаружения объектов на изображении? - PullRequest
0 голосов
/ 09 июля 2020

Я выполнил несколько руководств по обнаружению объектов с помощью TensorFlow. Мне удалось обучить модель на моем пользовательском наборе данных с помощью Colab и загрузить замороженный график вывода. Теперь я хочу использовать этот граф вывода на изображениях для обнаружения объектов. Из того, что я так долго искал, сначала мне нужно узнать имена тензор (не совсем уверен, что это значит). но код такой:

import tensorflow as tf

def load_graph(frozen_graph_filename):
    with tf.io.gfile.GFile(frozen_graph_filename, "rb") as f:
        graph_def = tf.compat.v1.GraphDef()
        graph_def.ParseFromString(f.read())
    with tf.Graph().as_default() as graph:
        tf.import_graph_def(graph_def, name="prefix")
    return graph
g = load_graph("inference_graph/frozen_inference_graph.pb")
for op in g.get_operations():
    print(op.values())

И это последние 6 строк:

(<tf.Tensor 'prefix/add/y:0' shape=() dtype=float32>,)
(<tf.Tensor 'prefix/add:0' shape=<unknown> dtype=float32>,)
(<tf.Tensor 'prefix/detection_boxes:0' shape=<unknown> dtype=float32>,)
(<tf.Tensor 'prefix/detection_scores:0' shape=<unknown> dtype=float32>,)
(<tf.Tensor 'prefix/detection_classes:0' shape=<unknown> dtype=float32>,)
(<tf.Tensor 'prefix/num_detections:0' shape=<unknown> dtype=float32>,)

Как мне продолжить? Я использую тензорный поток 2. Я пытался следовать некоторым руководствам, но они очень сбивают с толку. Я не могу заставить его работать так далеко.

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