opencv - 如何在 DIGITS（不同标签）中使用具有不同数据集的新预训练模型？

Question

我想使用 VGG_ILSVRC_19_layers 作为数字预训练模型，但具有不同的数据集。我需要不同的标签文件吗？如何上传此模型并将其用于我的数据集？

对于我得到的 VGG 16 层

错误：无法从层“fc6”复制参数 0 权重；形状不匹配。源参数形状为 1 1 4096 25088 (102760448)；目标参数形状为 4096 32768 (134217728)。要从头开始学习该层的参数，而不是从保存的网络中复制，请重命名该层。

如何修改图层？

Answer 1

您的标签与 DIGITS 中的数据集相关联 - 而不是您的模型。当您上传一个预训练的 VGG 模型时，您可能需要重命名您的最后一个内部产品层（请参阅此答案），以便您的模型可以在 N 个类而不是 1000 个上工作。