1.數(shù)據(jù)集的下載與轉(zhuǎn)換

1）我們?cè)趍nist數(shù)據(jù)集上做測(cè)試，MNIST handwritten digit database, Yann LeCun, Corinna Cortes and Chris Burge在這里下載這四個(gè)文件：

2）然后解壓生成了以下四個(gè)文件：

3）數(shù)據(jù)格式轉(zhuǎn)換

新建一個(gè)文本文檔，更改后綴為 .bat? 在里面復(fù)制以下code：?

..\..\Build\x64\Release\convert_mnist_data.exe ?..\..\data\mnist\train-images.idx3-ubyte?..\..\data\mnist\train-labels.idx1-ubyte?..\..\examples\mnist\mnist_train_lmdb??

echo.??

..\..\Build\x64\Release\convert_mnist_data.exe ?..\..\data\mnist\t10k-images.idx3-ubyte??..\..\data\mnist\t10k-labels.idx1-ubyte?..\..\examples\mnist\mnist_test_lmdb?

pause ?

其實(shí)第一個(gè)路徑就是 exe文件所在路徑， 后面兩個(gè)就是數(shù)據(jù)所在路徑，最后一個(gè)是輸出文件路徑

例如：

F:\caffe-master\caffe-master\Build\x64\Release\convert_mnist_data.exe E:\caffetest\mnist\mnist\train-images.idx3-ubyte E:\caffetest\mnist\mnist\train-labels.idx1-ubyte E:\caffetest\mnist\mnist\mnist_train_lmdb

echo.

F:\caffe-master\caffe-master\Build\x64\Release\convert_mnist_data.exe E:\caffetest\mnist\mnist\t10k-images.idx3-ubyte E:\caffetest\mnist\mnist\t10k-labels.idx1-ubyte E:\caffetest\mnist\mnist\mnist_test_lmdb

pause

保存后，雙擊運(yùn)行，生成了兩個(gè)文件夾：mnist_train_lmdb和mnist_test_lmdb

?2.打開lenet_train_test.prototxt

然后打開lenet_solver.prototxt,修改第二行即可，

可以將prototxt文件放在以下的網(wǎng)址，查看網(wǎng)絡(luò)的結(jié)構(gòu)：http://ethereon.github.io/netscope/#/editor

3.使用caffe.exe 訓(xùn)練網(wǎng)絡(luò)

利用命令行cd到caffe.exe的工作目錄，輸入以下的命令：

caffe train -solver=lenet_solver.prototxt -gpu 0

可以得到以下的訓(xùn)練結(jié)果：

4.使用Python調(diào)用caffe

環(huán)境：python+Anaconda；同時(shí)將編譯生成的python文件夾放在..\anaconda2\Lib\site-packages中。

import caffe caffe.set_mode_gpu() solver = caffe.SGDSolver('lenet_solver.prototxt') solver.solve()

運(yùn)行代碼可以得到以下的結(jié)果：

?

轉(zhuǎn)載于:https://www.cnblogs.com/youngsea/p/9430875.html

與50位技術(shù)專家面對(duì)面20年技術(shù)見證，附贈(zèng)技術(shù)全景圖

總結(jié)

以上是生活随笔為你收集整理的[caffe(一)]使用caffe训练mnist数据集的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

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