Keras 使用 ImageNet?

数据挖掘 机器学习 深度学习 喀拉斯
2022-02-15 20:52:32

Keras 提到它提供了在 ImageNet 上预训练的模型。但是,它没有具体说明“ImageNet”的含义——比如它是完整图像集的 ImageNet 的某个子集吗?我的意思是,我很难想象他们已经使用了所有 1400 万张图像进行预训练,但也许我错了。

我已经阅读了很多关于一些提出的架构的研究论文,它们都接受了 ImageNet 子集的训练,专门针对 ImageNet 大规模视觉识别挑战赛,这在论文中明确提到。当然,如果这些 Keras 权重实际上代表了整个 ImageNet,则没有问题 :-) 但需要某种确认/确认。

有谁更具体地知道“ImageNet”是什么意思?https://keras.io/api/applications/

1个回答

您提供的Keras链接本身提供了指向每个模型定义的链接。例如,ResNet-50相关论文位于: https ://arxiv.org/abs/1512.03385

该论文明确指出:

  1. 实验

4.1。ImageNet 分类

我们在包含 1000 个类的 ImageNet 2012 分类数据集 [36] 上评估我们的方法。这些模型在 128 万张训练图像上进行训练,并在 50k 验证图像上进行评估。我们还获得了测试服务器报告的 100k 测试图像的最终结果。我们评估前 1 和前 5 的错误率。

大多数进入 ImageNet 竞赛的模型都使用相同的子集,但需要得到每篇论文的确认。