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

Question

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

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

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

Answer 1

HDF5 (.h5, .hdf5) is a file format suitable for storing large collections of multidimensional numeric arrays (e.g. models, data files).

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

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

从这里开始，您最简单的选择就是您已经提到的:“从保存的模型中提取此信息并[out]需要使用预测函数”。