visualkeras: custom VGG example#

An example showing the visualkeras function used by a tf.keras.Model model.

# Authors: The scikit-plots developers
# SPDX-License-Identifier: BSD-3-Clause

pip install protobuf==5.29.4

import tensorflow as tf

# Clear any session to reset the state of TensorFlow/Keras
tf.keras.backend.clear_session()

from scikitplot import visualkeras

model = tf.keras.applications.VGG16(: include_top=True, weights=None, # “imagenet” or ‘path/’ input_tensor=None, input_shape=None, pooling=None, classes=1000, classifier_activation=”softmax”, name=”vgg16”,

) model.summary() visualkeras.layered_view(

model, legend=True, show_dimension=True, to_file=’result_images/vgg16.png’,

)

model = tf.keras.applications.VGG19(
    include_top=True,
    weights=None,  # "imagenet" or 'path/'
    input_tensor=None,
    input_shape=None,
    pooling=None,
    classes=1000,
    classifier_activation="softmax",
    name="vgg19",
)
# model.summary()

img_vgg19 = visualkeras.layered_view(
    model,
    legend=True,
    show_dimension=True,
    min_z=1,
    min_xy=1,
    max_z=4096,
    max_xy=4096,
    scale_z=0.5,
    scale_xy=11,
    font={"font_size": 199},
    # to_file="result_images/vgg19.png",
    save_fig=True,
    save_fig_filename="vgg19.png",
    overwrite=False,
    add_timestamp=True,
    verbose=True,
)
img_vgg19

[INFO] Saving path to: /home/circleci/repo/galleries/examples/visualkeras_CNN/result_images/vgg19_20251029_115819Z.png

<matplotlib.image.AxesImage object at 0x7fe91030b2d0>

Tags: model-type: classification model-workflow: model building plot-type: visualkeras domain: neural network level: beginner purpose: showcase

Total running time of the script: (0 minutes 15.106 seconds)

Related examples