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
Force garbage collection
import gc
gc.collect()
3
# 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,
)

[INFO] Saving path to: /home/circleci/repo/galleries/examples/visualkeras_CNN/result_images/vgg19_20250422_204317Z.png
[INFO] Image saved using Matplotlib: /home/circleci/repo/galleries/examples/visualkeras_CNN/result_images/vgg19_20250422_204317Z.png
Total running time of the script: (0 minutes 13.924 seconds)
Related examples

Visualkeras: Spam Classification Conv1D Dense Example
Visualkeras: Spam Classification Conv1D Dense Example