代码之家 › 专栏 › 技术社区 › hikaru

Keras应用模型的ImageNet测试数据集

imagenet keras

0

hikaru · 技术社区 · 7 年前

我有一个像这样在ImageNet上预先培训过的模型:

from keras.applications import resnet50
model = resnet50.ResNet50(weights='imagenet')

是否有任何方法可以获取测试ImageNet数据集及其标签(这意味着数据不用于上述培训) model )?

1 回复 | 直到 7 年前

1

0

Gaurav Gola 7 年前

可以加载权重为“无”和“无”的模型 download 然后,您可以将数据集拆分为列车测试,用列车组训练您的模型,并将测试集用于您的目的。