gpt4 book ai didi

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

转载 作者:太空宇宙 更新时间:2023-11-03 21:43:54 25 4
gpt4 key购买 nike

我想使用 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/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com