如何将训练好的 Caffe 模型用于当前输入图像？

Question

Caffe 的新手。

我正在尝试使用 Caffe 深度学习框架在 MNIST 数据集上使用经过训练的卷积神经网络。

关注official tutorial。

成功采取的步骤：

./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh
./examples/mnist/train_lenet.sh

模型已训练并停止并显示以下消息：

I1203 solver.cpp:133] Snapshotting solver state to lenet_iter_10000.solverstate
I1203 solver.cpp:78] Optimization Done.

现在，我不知道如何获取测试图像并使用现有的训练模型（我相信它是名称为 lenet_iter_10000.solverstate 的快照）来查看每个类别的预测分数。

Answer 1

使用caffe的test函数：

<path to caffe root>/caffe test -model <val filename>.prototxt -weights lenet_iter_10000.caffemodel

由于您只想测试一张图像，请将该图像作为输入提供给您的测试数据层。在您的<val filename>.protoxt 中也使用mean_image 作为输入。在这种情况下，测试批量大小为 1。

另请注意，lenet_iter_10000.solverstate 不是您经过训练的模型。您训练的模型实际上是lenet_iter_10000.caffemodel。要了解solverstate 和caffemodel 文件之间的差异，请参阅here。