绘制的Keras模型没有给出尺寸

2024-09-28 04:22:20 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图用Keras绘制VGG16架构图。但是,我看到了下面的图片,带有None,某种维度。None是什么意思,这是一个bug吗?在

enter image description here

from keras.utils import plot_model
model = VGG19(include_top=True, weights='imagenet')
#print(model.summary())
plot_model(model, to_file='model_Vgg19.png',show_shapes=True)

我可以看到这是Keras中的一个已知问题:https://github.com/keras-team/keras/issues/1877。在


Tags: fromimportnonetruemodelplot绘制图片
1条回答
网友
1楼 · 发布于 2024-09-28 04:22:20

shape tuple中的None维度是指批处理维度,这仅仅意味着层可以接受任何大小的输入。例如,一个数据集(例如MNIST)可以有以下形状元组(60000, 28, 28, 1),但是输入层的形状是(None, 28, 28, 1)。在

参考文献:

相关问题 更多 >

    热门问题