Я выполнил несколько руководств по обнаружению объектов с помощью 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. Я пытался следовать некоторым руководствам, но они очень сбивают с толку. Я не могу заставить его работать так далеко.