我想使用 VGG_ILSVRC_19_layers 作为数字预训练模型，但数据集不同。 我需要不同的标签文件吗？如何上传此模型并将其用于我的数据集？
对于我得到的VGG 16层
ERROR: Cannot copy param 0 weights from layer 'fc6'; shape mismatch. Source param shape is 1 1 4096 25088 (102760448); target param shape is 4096 32768 (134217728). To learn this layer's parameters from scratch rather than copying from a saved net, rename the layer.
如何修改图层？
最佳答案
您的标签与 DIGITS 中的数据集相关联，而不是您的模型。当您上传预训练的 VGG 模型时，您可能需要重命名最后一个内积层(请参阅 this answer)，以便您的模型可以处理 N 个类而不是 1000 个。
关于opencv - 如何在 DIGITS(不同标签)中使用具有不同数据集的新预训练模型？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/40121099/