是否可以从保存的 CNN 中提取分配给训练图像的类别概率？

Question

我训练了一个 CNN 来执行分类任务，我使用了 vgg16 和 categorical_crossentropy 损失函数、sgd 优化器和 softmax 激活函数。

然后我将训练好的模型保存到 h5 文件中。 我知道获取新的看不见的图像的类概率向量的函数是predict。

1. 但是，我想知道分配给训练图像的最终类别概率是否与模型一起保存？
2. 如果是，我如何在不需要使用predict 函数的情况下从保存的模型中提取此信息？

Answer 1

HDF5 (.h5, .hdf5)是一种适合存储大容量的文件格式 多维数值数组的集合（例如模型、数据 文件）。

1. 是的，您可以在训练期间将“分配给训练图像的最终类别概率”保存到 .h5 文件中。

2. 但由于您只“将训练后的模型保存到 .h5 文件”，因此将只有网络模型（如果您定义了它的权重、架构和优化器配置） ) 在此文件中（更多信息可以在here - “作为 HDF5 文件”部分）中找到。

从这里你最简单的选择就是你已经提到的：“从保存的模型中提取此信息[out]需要使用预测功能”。