用Caffe 訓練和測試MNIST資料

2021-08-08 02:03:39 字數 1140 閱讀 4368

c

affe安裝包自帶mnist的例子。測試步驟如下

1.獲得mnist的資料報,在caffe的根目錄下執行./date/mnist/get_mnist.sh指令碼

2.生成lmdb

執行./example/mnist/create_mnist.sh。將mnist date 轉化成caffe可用的lmdb格式檔案。

3. 定義mnist network

3.1 寫data layer

目前我們從之前創立的lmdb中讀取mnist data,該資料層定義為

layers

top: "data"

top: "label"

}3.2 寫convolution layer

定義地一層convolution layer:

layers

bias_filler  

}bottom: "data"

top: "conv1"}

3.3 寫pooling layer

layers

bottom: "conv1"

top: "pool1"

}類似地,你還可以定義第二個covolution和pooling layers, 具體可參考

/examples/mnist/lenet_train_test.prototxt

3.4 寫fully connected layer

layers

bias_filler

}bottom: "pool2"

top: "ip1"

} 3.5 寫relu layer

layers

在relu layer之後,我們可以寫另乙個innerproduct layer:

layers

bias_filler

}bottom: "ip1"

top: "ip2"  

}3.6 寫loss layer

最後寫loss

layers

4. 訓練網路  執行./example/mnist/train_lenet.sh 5.

測試網路

執行./example/mnist/test_lenet.sh

測試資料中的accruacy的平均成功率0.99

windows下caffe之訓練mnist

一 參考資料 悠望南山 caffe 學習筆記 一 caffe example 之訓練mnist 二 訓練所需要用到的檔案 1.mnist test lmdb 測試資料集,裡面包括data.mdb和lock.mdb 2.mnist train lmdb 訓練資料集,裡面包括data.mdb和lock....

Ubuntu安裝Caffe並訓練MNIST

ubuntu安裝caffe並訓練mnist。今天介紹一下如何在ubuntu下安裝caffe並且訓練mnist。當然,第一手的資料同樣來自於caffe官網 版本是16.04,caffe是cpu only的版本。caffe安裝 通過下面的命令安裝protobuf,leveldb,snappy,openc...

用訓練好的caffe模型來測試樣本

如果要把訓練好的模型拿來測試新的樣本,那必須得要乙個deploy.prototxt檔案,這個檔案實際上和test.prototxt檔案差不多,只是頭尾不相同而已。deploy檔案沒有第一層資料輸入層,樣本是被直接輸入到net.blobs裡面的,也沒有最後的accuracy層,但最後多了乙個softm...