Tensorflow显示模型摘要

Question

我有一个经过培训并保存了模型的UNet。我想查看模型摘要，但是当我使用model.summary()时，Output Shape列被截断并且看不到完整的形状。模型是在3D图像上训练的，因此输出应显示(None, shapeX, shapeY, shapeZ, num_features)。如何显示完整的输出形状？

from tensorflow.keras.models import load_model
        
model = load_model('model.h5',compile = False)
model.summary()

model.summary（）

Answer 1

您可以设置line_length函数的tf.summary属性。

model.summary(line_length = 100)