Wie findet man die Input-und Output-Knoten eine Gefrorene Modell
Möchte ich tensorflow ist optimize_for_inference.py
Skript auf einem gefrorenen Modell aus der Modell-zoo: die ssd_mobilenet_v1_coco
.
Wie finde ich/bestimmen Sie die Namen der input-und output-Namen des Modells?
Hier ist ein link zu dem Diagramm generiert tensorboard
Dieser Frage helfen könnte: Gegeben ein tensor-flow-Modell-Diagramm, so finden Sie den Eingabe-Knoten und Ausgangs-Knoten-Namen (für mich nicht)
Du musst angemeldet sein, um einen Kommentar abzugeben.
Wir haben nun eine Dokumentation über diesen Prozess, die hier behandelt werden:
https://www.tensorflow.org/mobile/prepare_models
Wenn Sie sich für summarize_graph, sehen Sie ein Beispiel, wie um herauszufinden, die richtige input-und output-Knoten.
bazel build tensorflow/tools/graph_transforms:summarize_graph
bazel-bin/tensorflow/tools/graph_transforms/summarize_graph --in_graph=tensorflow_inception_graph.pb
oderpython import_pb_to_tensorboard.py --model_dir <model path> --log_dir <log dir path>