繁体   English   中英

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

[英]VGG16 Training new dataset: Why VGG16 needs label to have shape (None,2,2,10) and how do I train mnist dataset with this network?

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

(我使用的是keras,Python3.5)

层次问题导致精度低。 我刚修改了我的网络并获得了.7496的准确性。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM