Wie können Layer hinzugefügt werden, bevor die Eingabeebene des Modells aus einer .pb-Datei wiederhergestellt wird?

Question

Ich habe eine .pb-Modell-Datei und ich kann den Eingabe Tensor und Ausgabe Tensor von get_tensor_by_name und dann den Eingabe-und Ausgabe-Tensor, um ein gespeichertes Modell erstellen, die dann in Tensorflow-Modell-Server serviert werden können. Aber im Moment ist der Eingangstensor Bilder im 3D-Array-Format, und ich möchte eine weitere Ebene hinzufügen, um jpeg-codierte Strings zu dekodieren und das Ergebnis in den aktuellen Eingangstensor einzufügen. Irgendwelche Vorschläge, wie das geht?

Answer 1

Sie können diese Ebenen hinzufügen, indem Sie die Datei export_inference_graph hier https://github.com/tensorflow/models/blob/master/research/slim/export_inference_graph.py#L114-L116 bearbeiten.

Sie können hier eine beliebige Auswertung des Vorverarbeitungstensors hinzufügen, bevor Sie das Inferenzdiagramm nach dem Platzhaltercode exportieren.