我试图用Keras绘制VGG16架构图。但是,我看到了下面的图片,带有None
,某种维度。None
是什么意思,这是一个bug吗?在
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。在
shape tuple中的
None
维度是指批处理维度,这仅仅意味着层可以接受任何大小的输入。例如,一个数据集(例如MNIST)可以有以下形状元组(60000, 28, 28, 1)
,但是输入层的形状是(None, 28, 28, 1)
。在参考文献:
相关问题 更多 >
编程相关推荐