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

Question

我正在尝试使用Keras绘制VGG16架构图。但是，我看到下面带有None的图片，有些尺寸。 None的含义是什么，是一个错误？

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。

Answer 1

形状元组中的None维度是指批量维度，这意味着图层可以接受任何大小的输入。例如，数据集（例如MNIST）可以具有以下形状元组(60000, 28, 28, 1)，但输入图层的形状将为(None, 28, 28, 1)。

参考文献：

• keras-team/keras#2054
• Keras Documentation - Core Layers