VGG16培训新数据集:为什么VGG16需要标签具有形状(无,2,2,10)以及如何使用此网络训练mnist数据集?

时间:2017-06-23 14:08:13

标签: python machine-learning deep-learning keras

我试图在VGG16网络上训练CIFAR10和MNIST数据集。在我的第一次尝试中,我得到一个错误,其中input_2(标签)的形状必须是(None,2,2,10)。这个结构在2x2x10数组中保存了什么信息,因为我希望input_2具有形状(无,10)(我的数据集中有10个类)。 我试图将标签的尺寸从(无,10)扩展到(无,2,2,10)。但我确信这不是正确的方法,因为我获得了非常低的准确度(大约0.09)

(我正在使用keras,Python3.5)

1 个答案:

答案 0 :(得分:0)

低精度是由层中的问题引起的。我刚修改了我的网络并获得了.7496的准确度。