web-dev-qa-db-fra.com

Comment représenter graphiquement le modèle tf.keras dans Tensorflow-2.0?

J'ai mis à niveau vers Tensorflow 2.0 et il n'y a pas de tf.summary.FileWriter("tf_graphs", sess.graph). Je regardais d'autres questions sur StackOverflow à ce sujet et ils ont dit d'utiliser tf.compat.v1.summary etc. Il doit sûrement y avoir un moyen de représenter graphiquement et de visualiser un modèle tf.keras dans Tensorflow version 2. Qu'est-ce que c'est? Je recherche une sortie de tensorboard comme celle ci-dessous. Je vous remercie!

enter image description here

python-3.xtensorflowtensorboardtensorflow2.0tf.keras
11
Colin Steidtmann

Une autre option consiste à utiliser ce site Web: https://lutzroeder.github.io/netron/

qui génèrent un graphique avec un fichier .h5 ou .tflite.

Le dépôt github sur lequel il est basé peut être trouvé ici: https://github.com/lutzroeder/netron

0
driedler

Voici ce qui fonctionne pour moi en ce moment (TF 2.0.0), basé sur le code tf.keras.callbacks.TensorBoard :

# After model has been compiled
from tensorflow.python.ops import summary_ops_v2
from tensorflow.python.keras.backend import get_graph
tb_path = '/tmp/tensorboard/'
tb_writer = tf.summary.create_file_writer(tb_path)
with tb_writer.as_default():
    if not model.run_eagerly:
        summary_ops_v2.graph(get_graph(), step=0)
0
K A