caffe for windows 訓練cifar10

查志強發表於2015-07-24

【原文:http://www.th7.cn/system/win/201411/77318.shtml

1 cifar10資料庫

60000張32*32 彩色圖片 共10類

50000張訓練

10000張測試

下載cifar10資料庫:http://www.cs.toronto.edu/~kriz/cifar-10-binary.tar.gz

這是binary格式的,所以我們要把它轉換成leveldb格式。



2 在../caffe-windows/examples/cifar10資料夾中有一個 convert_cifar_data.cpp

將他include到MainCaller.cpp中。如下:


 

編譯....我是一次就通過了 ,在bin資料夾裡出現convert_cifar_data.exe。然後 就可以進行格式轉換。binary→leveldb

可以在bin資料夾下新建一個input資料夾。將cifar10.binary檔案放在input資料夾中,這樣轉換時就不用寫路徑了。

cmd進入bin資料夾


執行後,在output資料夾下有cifar_train_leveldb和cifar_test_leveldb兩個資料夾。裡面是轉化好的leveldb格式資料。

當然,也可以寫一個bat檔案處理,方便以後再次使用。



3 下面我們要求資料影象的均值

編譯../../tools/comput_image_mean.cpp


編譯成功後,沒有出現comput_image_mean.exe。沒關係,我們還有maincaller.exe。接下來求mean

cmd進入bin。

執行後,在bin資料夾下出現一個mean.binaryproto檔案,這就是所需的均值檔案。


4 訓練cifar網路

在.../examples/cifar10資料夾裡已經有網路的配置檔案,我們只需要將cifar_train_leveldb和cifar_test_leveldb兩個資料夾還有mean.binaryproto檔案拷到cifar0資料夾下。

修改cifar10_quick_train.prototxt中的source: "cifar-train-leveldb"    mean_file: "mean.binaryproto" 和cifar10_quick_test.prototxt中的source: "cifar-test-leveldb"
    mean_file: "mean.binaryproto"就可以了,

後面再訓練就類似於MNIST的訓練。寫一個train_quick.bat,內容如下:

copy ..//..//bin//MainCaller.exe ..//..//bin//train_net.exeSET GLOG_logtostderr=1"../../bin/train_net.exe" cifar10_quick_solver.prototxt  pause

先編譯一遍 train_net.cpp

執行train_quick.bat

  


5 執行結果




1 cifar10資料庫

60000張32*32 彩色圖片 共10類

50000張訓練

10000張測試

下載cifar10資料庫:http://www.cs.toronto.edu/~kriz/cifar-10-binary.tar.gz

這是binary格式的,所以我們要把它轉換成leveldb格式。



2 在../caffe-windows/examples/cifar10資料夾中有一個 convert_cifar_data.cpp

將他include到MainCaller.cpp中。如下:


 

編譯....我是一次就通過了 ,在bin資料夾裡出現convert_cifar_data.exe。然後 就可以進行格式轉換。binary→leveldb

可以在bin資料夾下新建一個input資料夾。將cifar10.binary檔案放在input資料夾中,這樣轉換時就不用寫路徑了。

cmd進入bin資料夾


執行後,在output資料夾下有cifar_train_leveldb和cifar_test_leveldb兩個資料夾。裡面是轉化好的leveldb格式資料。

當然,也可以寫一個bat檔案處理,方便以後再次使用。



3 下面我們要求資料影象的均值

編譯../../tools/comput_image_mean.cpp


編譯成功後,沒有出現comput_image_mean.exe。沒關係,我們還有maincaller.exe。接下來求mean

cmd進入bin。

執行後,在bin資料夾下出現一個mean.binaryproto檔案,這就是所需的均值檔案。


4 訓練cifar網路

在.../examples/cifar10資料夾裡已經有網路的配置檔案,我們只需要將cifar_train_leveldb和cifar_test_leveldb兩個資料夾還有mean.binaryproto檔案拷到cifar0資料夾下。

修改cifar10_quick_train.prototxt中的source: "cifar-train-leveldb"    mean_file: "mean.binaryproto" 和cifar10_quick_test.prototxt中的source: "cifar-test-leveldb"
    mean_file: "mean.binaryproto"就可以了,

後面再訓練就類似於MNIST的訓練。寫一個train_quick.bat,內容如下:

copy ..//..//bin//MainCaller.exe ..//..//bin//train_net.exeSET GLOG_logtostderr=1"../../bin/train_net.exe" cifar10_quick_solver.prototxt  pause

先編譯一遍 train_net.cpp

執行train_quick.bat

  


5 執行結果



相關文章