Ubuntu 14.04 64位元機上用Caffe+MNIST訓練Lenet網路操作步驟

2020-06-16 17:50:06

Ubuntu 14.04 64位元機上用Caffe+MNIST訓練Lenet網路操作步驟

1.將終端定位到Caffe根目錄；

2.下載MNIST資料庫並解壓縮：$ ./data/mnist/get_mnist.sh

3.將其轉換成Lmdb資料庫格式：$ ./examples/mnist/create_mnist.sh

執行完此shell指令碼後，會在./examples/mnist下增加兩個新目錄，mnist_test_lmdb和mnist_train_lmdb

4.train model：$ ./examples/mnist/train_lenet.sh

(1)、使用LeNet網路(《Gradient-BasedLearning Applied to Document Recognition》)；

(2)、使用./examples/mnist/lenet_train_test.prototxtmodel；

(3)、使用./examples/mnist/lenet_solver.prototxtmodel；

(4)、執行train_lenet.sh指令碼，會呼叫./build/tools目錄下的caffe執行檔案，此執行檔案的實現是./tools目錄下的caffe.cpp檔案；

(5)、執行此指令碼後，會生成幾個檔案，其中./examples/mnist/lenet_iter_10000.caffemodel則是最終訓練生成的model檔案；

(6)、以上預設的是在GPU模式下執行，如果想讓其在CPU模式下執行，只需將lenet_solver.prototxt檔案中的solver_mode欄位值由原來的GPU改為CPU即可；

執行結果如下圖：

參考文獻：http://caffe.berkeleyvision.org/gathered/examples/mnist.html