是否可以从保存的 CNN 中提取分配给训练图像的 类 概率?
Is it possible to extract classes probabilities assigned to training images from a saved CNN?
我训练了一个 CNN 来执行分类任务,我使用了 vgg16
和 categorical_crossentropy
损失函数、sgd
优化器和 softmax
激活函数。
然后我将训练好的模型保存到h5文件中。
我知道获取新的未见过图像的 类 概率向量的函数是 predict
。
- 但是,我想知道分配给训练图像的最终 类 概率是否与模型一起保存?
- 如果是,我如何在不使用
predict
函数的情况下从保存的模型中提取此信息?
HDF5 (.h5, .hdf5)
is a file format suitable for storing large
collections of multidimensional numeric arrays (e.g. models, data
files).
是的,您可以在训练期间将 "final classes probabilities assigned to training images" 保存到 .h5
文件中。
但是因为你只 "saved then the trained model to .h5
file" 将只有网络模型(如果你这样定义它的权重,架构和优化器配置) 在此文件中(可以找到更多信息 here - "As an HDF5 file" 部分)。
从这里开始,您最简单的选择就是您已经提到的:"Extract this information from the saved model with[out] the need of using predict function".
我训练了一个 CNN 来执行分类任务,我使用了 vgg16
和 categorical_crossentropy
损失函数、sgd
优化器和 softmax
激活函数。
然后我将训练好的模型保存到h5文件中。
我知道获取新的未见过图像的 类 概率向量的函数是 predict
。
- 但是,我想知道分配给训练图像的最终 类 概率是否与模型一起保存?
- 如果是,我如何在不使用
predict
函数的情况下从保存的模型中提取此信息?
HDF5 (.h5, .hdf5)
is a file format suitable for storing large collections of multidimensional numeric arrays (e.g. models, data files).
是的,您可以在训练期间将 "final classes probabilities assigned to training images" 保存到
.h5
文件中。但是因为你只 "saved then the trained model to
.h5
file" 将只有网络模型(如果你这样定义它的权重,架构和优化器配置) 在此文件中(可以找到更多信息 here - "As an HDF5 file" 部分)。
从这里开始,您最简单的选择就是您已经提到的:"Extract this information from the saved model with[out] the need of using predict function".