caffe练习实例(1)——训练mnist数据集

xiaoxiao2021-02-28  104

1.获取数据 cd  caffe/data/mnist 执行命令:./get_mnist.sh 执行完命令后切换到caffe/data/mnist目录下发现get_mnist.sh文件是原有的,其他四个文件是下载的。如图1所示

                                  图1

2.将数据转化为lmdb格式 命令:./examples/mnist/create_mnist.sh    切记必须在caffe的根目录下运行否则报错

3.训练

命令:./examples/mnist/train_lenet.sh 切记也必须在caffe的根目录下运行否则报错 4.结果如下图2所示                                                                  图2 用GPU训练的一分钟不到就跑完了,准确率99.02%比网上说的好像低了点。 5.如果你没有gpu那就用cpu跑的话可能20分钟左右 执行以下命令修改CPU还是GPU跑:cd caffe/examples/mnist/ vim lenet_solver.prototxt 然后修改成GPU或者CPU即可,如下图3所示                          图3

转载请注明原文地址: https://www.6miu.com/read-32129.html

最新回复(0)