我可以找到训练 tensorflow 预训练模型的类名吗?

时间:2021-07-12 09:01:03

标签: tensorflow machine-learning keras pre-trained-model

我正在使用以下数据集:

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

但我不知道如何从预训练模型中获取类名。有什么方法或任何声明可以做到这一点吗?

0 个答案:

没有答案