对于caffe训练mnist需要什么样的数据集,预处理成什么样子,在这里不再说(=@__@=)了?网上有很多哦!本次主要讲如何训练,测试自己网络准确率和识别自己的手写图片手写数字训练网络
训练运行:
y@y312:~/caffe-master$ ./examples/mnist/train_lenet.sh
这个是通过脚本.sh 的方式运行、
也可以使用sudo build/tools/caffe train -solver=examples/xx/xxx.solver.prototxt 的方式运行
caffe 的均值计算
sudo build/tools/compute_image_mean examples/mnist/mnist_train_lmdb examples/mnist/mean.binaryproto
使用训练好的model 进行测试手写字体图片识别
y@y312:~/caffe-master$ sudo ./build/examples/cpp_classification/classification.bin examples/mnist/lenet.prototxt examples/mnist/lenet_iter_10000.caffemodel examples/mnist/mean.binaryproto examples/mnist/label.txt examples/mnist/0.png
训练好的model进行验证集批量测试网络的准确率
./build/tools/caffe.bin test -model examples/mnist/lenet_train_test.prototxt -weights examples/mnist/lenet_iter_10000.caffemodel -iterations 100
对于其他实例的网络以及其他数据的训练与测试都是这样的方法,可以思考思考哦~
caffe 初步训练与测试
最新推荐文章于 2019-08-23 18:40:03 发布