визуализация моделей с использованием plot_model tensorflow.keras

Question

>>> from tensorflow.keras.applications import VGG16
>>> from tensorflow.keras.utils import plot_model
>>> plot_model(VGG16, to_file="vgg16.png", show_shapes=True)

AttributeError: 'function' object has no attribute '_is_graph_network'

Я попытался изменить show_shapes = False, но появилась та же ошибка. Мне удалось запустить программу для моделей, сделанных на заказ.

Answer 1

tensorflow.keras.applications.VGG16 - это функция, которую нужно вызвать для получения экземпляра keras.Model. См. https://www.tensorflow.org/api_docs/python/tf/keras/applications/VGG16.

Что-то, что подойдет:

from tensorflow.keras.applications import VGG16
from tensorflow.keras.utils import plot_model

plot_model(VGG16())