Блог

У меня есть 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()

Вы можете установить line_length свойство tf.summary функции.

 model.summary(line_length = 100)