CIFAR-10在caffe上進行訓練

        CIFAR-10數據集含有6萬個32*32的彩色圖像,共分爲10種類型,由 AlexKrizhevsky, Vinod Nair和 Geoffrey Hinton收集而來。包含50000張訓練圖片,10000張測試圖片。

        這是該數據集的一些圖片。


下面爲具體步驟

第一步,在終端運行以下指令,下載CIFAR-10數據庫。

cd $CAFFE_ROOT/data/cifar10
./get_cifar10.sh


第二步,然後進入目錄 執行

cd $CAFFE_ROOT
./examples/cifar10/create_cifar10.sh

可以打開create_cifar10.sh文件,查看具體做了什麼。其實就是將圖片庫轉成leveldb格式,並計算均值二進制文件。執行之後,可以發現examples文件夾下中出現數據庫文件cifar10-leveldb和數據庫圖像均值二進制文件mean.binaryproto。

 

 

第三步,訓練cifar網絡。

如果不支持GPU訓練,則需要將CAFFE_ROOT/examples/cifar10/cifar10_quick_solver.prototxt中的最後一行改爲solver_mode:CPU,如下圖:


如果支持GPU訓練,則不需要改。

 

接下來,回到caffe目錄,運行train_quick.sh。

cd $CAFFE_ROOT
./examples/cifar10/train_quick.sh


運行之後,可以看到終端便開始打印出一些信息。

部分信息如下:可以看到到迭代2500次時,準確率爲0.6858.



注1:也可以看看$CAFFE_ROOT /examples/cifar10文件夾下的readme.md文件,裏面也有一些步驟說明。

 

注2:第二步要是先進入$CAFFE_ROOT /examples/cifar10/目錄,再運行create_cifar10.sh,會出現如下錯誤:


解決方法是要在$CAFFE_ROOT目錄下運行create_cifar10.sh文件。









發佈了43 篇原創文章 · 獲贊 105 · 訪問量 36萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章