首页 > 解决方案 > 我可以找到训练 tensorflow 预训练模型的类名吗?

问题描述

我正在使用以下数据集:

https://www.kaggle.com/abdallahalidev/plantvillage-dataset

我有一个预先训练好的模型。我通过以下方式评估了模型:

from tensorflow.keras.models import load_model
model=load_model('plantvillage.h5')
model.evaluate_generator(test_image_gen)
model.metrics_names

但是我无法弄清楚如何从预训练模型中获取类名。有没有办法或任何声明来做到这一点?

标签: tensorflowmachine-learningkeraspre-trained-model

解决方案


推荐阅读