運(yùn)行caffe自帶的mnist實(shí)例教程
1、先進(jìn)入caffe文件目錄，(指令：cd ./caffe)，再用data/mnist下的get_mnist.sh下載MNIST數(shù)據(jù)集，代碼如下：

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

2、轉(zhuǎn)換格式，代碼如下：

./examples/mnist/create_mnist.sh

3、訓(xùn)練超參數(shù)，有兩種方式：

• （1）通過命令行執(zhí)行訓(xùn)練，代碼如下：（這是書上的例子，可以參考網(wǎng)址上的例子，在后面。不同之處：網(wǎng)址上的例子在此處新建一個(gè)文件夾保存訓(xùn)練的模型）

./build/tools/caffe train --solver=examples/mnist/lenet_solver.prototxt

• 2）將以上訓(xùn)練的命令行代碼寫成訓(xùn)練腳本并命名為train_lenet.sh，放到mnist目錄下。運(yùn)行train_lenet.sh腳本進(jìn)行訓(xùn)練。

./examples/mnist/train_lenet.sh

5、用訓(xùn)練好的模型對數(shù)據(jù)進(jìn)行預(yù)測。

./build/tools/caffe.bin test -model=examples/mnist/lenet_train_test.prototxt -weights=examples/mnist/lenet_iter_5000.caffemodel -iterations=100

總結(jié)

以上是生活随笔為你收集整理的运行caffe自带的mnist实例教程的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。