使用plot_model

时间:2020-10-13 02:43:21

标签: python python-3.x tensorflow tensorflow2.0

使用tensorflow版本2.3.0时,输出模型不清楚,如何解决?

import tensorflow as tf

base_model = tf.keras.applications.EfficientNetB7(weights='imagenet', include_top=False)  # or weig
tf.keras.utils.plot_model(base_model, to_file="image1.png", dpi=120)

它看起来像下面的

enter image description here

1 个答案:

答案 0 :(得分:1)

问题是,您尝试输出为图像的模型是巨大的模型。因此,dpi=120仅以PNG格式不够用。为了获得更清晰的图像,请将其另存为PDF。然后,您可以根据需要轻松地转换为图像。

使用以下内容:

import tensorflow as tf

base_model = tf.keras.applications.EfficientNetB7(weights='imagenet', include_top=False)  # or weig
tf.keras.utils.plot_model(base_model, to_file="image1.pdf", dpi=120)

请给我以下内容(当我将屏幕打印在特定块上时):

enter image description here