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

11

別のオプションはこのウェブサイトを使用することです: 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