caffe訓練流程

2021-08-18 14:33:23 字數 1446 閱讀 9048

l  準備訓練資料流程如下

用指令碼生成好label.txt標籤檔案,格式如下(檔名+空格+標籤)

把所有放在乙個目錄下後cd到執行如下命令

convert_imageset  [flags] rootfolder/ listfile db_name  

其中db_name後面還可以跟一些可選的引數設定,具體有哪些可選的引數參見「可選引數設定部分」

其中rootfolder為影象集的根目錄

listfile 為乙個檔案的路徑,該檔案中記錄了影象集中的各圖樣的路徑和相應的標註

db_name為要生成的資料庫的名字

例:convert_imageset c:\\data\\pics_test\\ c:\\matchdata\\test.txt c:\\ data\\test_lmdb --gray=false--resize_width=32 --resize_height=32

這樣就在指定目錄下生成了 訓練資料檔案

生成驗證集檔案同上

生成均值檔案

cd到執行如下命令

compute_image_mean[flags] input_db [output_file]\n」) 

引數:input_db: 資料庫 

引數(可選):output_file: 輸出檔名,不提供的話,不儲存平均影象blob

例:compute_image_mean c:/matchdata/train_lmdbc:/matchdata/train_lmdb/mean.binaryproto

準備好solver超參檔案,train_val模型檔案,檔案內容涉及相關路徑的需要做相應替換

cd到執行如下命令

caffe train -solver c:/caffe/cifar10/ solver.prototxt

如果是用現有引數微調則還需要準備相應的.caffemodel檔案

執行caffe train -solver c:/caffe/cifar10/ solver.prototxt

-weights c:/caffe/cifar10/test.caffemodel

l  測試模型

cd到執行如下命令

其中deploy.prototxt 是根據train_val.protxt改寫而成(例檔案是訓練生成的引數檔案,synset_words.txt為標籤檔案,cat.jpg為測試

caffe訓練資料流程

cifar10訓練例項 sudo sh data cifar10 get cifar10.sh 2.轉換資料格式為lmdb sudo sh examples cifar10 create cifar10.sh 轉換成功後,會在 examples cifar10 資料夾下生成兩個資料夾,cifar10...

caffe訓練分類完整流程

caffe均值檔案mean.binaryproto轉mean.npy 深度學習 十三 caffe之訓練資料格式 caffe在cifar10上覆現resnet 使用caffe訓練自己的資料 使用caffe fine tune乙個單標籤影象分類模型 如何用caffe訓練影象分類深度學習模型 caffe下...

Caffe 訓練車輛 Reid

1 每個資料夾下保證是同一輛車,最好有多角度的變化,以便網路提取不同方向特徵 2 標籤 2 編譯caffe 注意對opencv python版本的要求,這個問題很無解,照做就是了 我的機器環境是opencv3.4原始碼編譯安裝 系統自帶py2.7,不推薦anaconda,opencv4.1 pip安...