caffe 训练时我们会在log中看到accuracy和loss,这里的accuracy是在val集上得到的。
在训练完得到模型后如果我们想在新的数据上测试并得到accuracy, loss改怎么做?
可以利用trainval.prototxt 文件稍作修改得到一个test.prototxt
修改的地方是Datalayer, 将source 改为测试数据的路径
layer { name: "cifar" type: "Data" top: "data" top: "label" include { phase: TEST } transform_param { mean_file: "new_mean.binaryproto" } data_param { source: "test_lmdb" batch_size: 100 backend: LMDB } }
然后利用shell(windows bat)文件
D:caffe-windowsBuildx64Releasecaffe.exe test -model test.prototxt -weights cifar10_100000.caffemodel -iterations 100