ubuntu18.04+caffe+MNIST例程测试

最新推荐文章于 2022-01-12 18:11:54 发布

Be_Patient

最新推荐文章于 2022-01-12 18:11:54 发布

阅读量926

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/sinat_41789705/article/details/82778580

本文介绍如何使用Caffe框架下载并处理MNIST数据集，通过将二进制文件转换为LMDB格式，最终训练LeNet模型实现手写数字识别，准确率达到99.1%。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.下载mnist数据集

 cd /caffe/data/mnist
 ./get_mnist.sh

2.将下载的二进制文件准换成LMDB文件

  cd caffe
  ./examples/mnist/creat_mnist.sh

3.测试

 cd caffe
./examples/mnist/train_lenet.sh

4.结果:
Initializing solver from parameters:
test_iter: 100
test_interval: 500
base_lr: 0.01
display: 100
max_iter: 10000
lr_policy: “inv”
gamma: 0.0001
power: 0.75
momentum: 0.9
weight_decay: 0.0005
snapshot: 5000
snapshot_prefix: “examples/mnist/lenet”
solver_mode: GPU
device_id: 0
net: "examples/mnist/lenet_train_test.prototxt
…