python – 保存中的问题 – 加载apollocaffe模型

我使用以下命令保存我训练的模型:

net.save(model_name)

然后使用以下命令加载它:

net.load(model_name)

但是在加载模型之后,当我尝试使用net.blobs()打印blob时,它给了我一个空字典.看起来该模型未正确保存或未正确加载.

请在这方面帮助我.提前致谢.

最佳答案 你的问题有两个要素.

保存:

要加载模型,必须将其保存为HDF5格式.这可以使用SnapshotLogger类完成,如下所示:

from apollocaffe.loggers import SnapshotLogger

SnapshotLogger(snapshot_interval, snapshot_prefix='PATH_TO_YOUR_MODEL',
        log_file="PATH_TO_log.txt")

加载:

Loading保存的模型可以如下所示完成:

net = apollocaffe.ApolloNet()
model_path = "../model_name.h5"
net.load(model_path)
点赞