web-dev-qa-db-fra.com

Comment trouver les nœuds d'entrée et de sortie d'un modèle gelé

Je veux utiliser le script optimize_for_inference.py De tensorflow sur un modèle figé du zoo modèle: le ssd_mobilenet_v1_coco.

Comment trouver/déterminer les noms des noms d'entrée et de sortie du modèle?

Voici un lien vers le graphique généré par tensorboard

Cette question pourrait aider: Étant donné un graphique de modèle d'écoulement tensoriel, comment trouver les noms des nœuds d'entrée et de sortie (pour moi, ce n'est pas le cas)

12
gustavz

Nous avons maintenant ajouté de la documentation sur ce processus, couverte ici:

https://www.tensorflow.org/mobile/prepare_models

Si vous recherchez resume_graph, vous verrez un exemple de la façon de déterminer les bons nœuds d'entrée et de sortie.

6
Pete Warden