web-dev-qa-db-ja.com

Tensorflow-2.0でtf.kerasモデルをグラフ化する方法は？

Tensorflow 2.0にアップグレードしましたが、tf.summary.FileWriter("tf_graphs", sess.graph)はありません。私はこれについて他のいくつかのStackOverflow質問を調べていましたが、彼らはtf.compat.v1.summary etc。 Tensorflowバージョン2でtf.kerasモデルをグラフ化して視覚化する方法が必ずあるはずです。それは何ですか？以下のようなテンソルボード出力を探しています。ありがとうございました！

enter image description here

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

別のオプションはこのウェブサイトを使用することです： https://lutzroeder.github.io/netron/

.h5または.tfliteファイルでグラフを生成します。

それが基づいているgithubリポジトリはここにあります： https://github.com/lutzroeder/netron

0
driedler

tf.keras.callbacks.TensorBoard コードに基づいて、現時点で私のために機能しているもの（TF 2.0.0）は次のとおりです。

# 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