caffe生成lmdb檔案

2021-09-17 22:37:00 字數 1173 閱讀 6053

從txt標籤檔案生成lmdb(其實是得到乙個資料夾,裡面有2個檔案)

current_path=`cd

$(dirname $0)

;pwd

` #獲取指令碼所在路徑

example=

$current_path

#lmdb要放的路徑

data=

$current_path

#txt檔案所在路徑

tools=/home/cdli/caffe-ssd/build/tools

train_data_root=/

val_data_root=/

# 這裡我們開啟resize,需要把所有尺寸統一

resize=true

if$resize

;then

resize_height=256

resize_width=256

else

resize_height=0

resize_width=0

fi..

.....

echo

"creating train lmdb..."

glog_logtostderr=1 $tools/convert_imageset \

--resize_height=

$resize_height \

--resize_width=

$resize_width \

--shuffle \

$train_data_root \

$data/train.txt \

$example/train_lmdb #生成的lmdb路徑

echo

"creating val lmdb..."

glog_logtostderr=1 $tools/convert_imageset \

--resize_height=

$resize_height \

--resize_width=

$resize_width \

--shuffle \

$val_data_root \

$data/val.txt \

$example/val_lmdb #生成的lmdb路徑

echo

"done."

生成caffe所用的lmdb資料

本文只介紹在caffe框架下生成lmdb格式的資料,因為在caffe框架下訓練和測試都只能用lmdb格式的資料 準備 1 ubuntu 18.04 2 新建data資料夾,內含train和test兩個資料夾 3 train資料夾和test資料夾裡面分別含img和xml兩個資料夾,裡面分別是原影象 j...

python讀取lmdb檔案 LMDB檔案讀取器

函式 tf.lmdbreader lmdbreader 類 繼承自 readerbase 從 lmdb 檔案中輸出記錄的讀取器.有關支援的方法,請參見 readerbase.屬性reader ref 實現讀取器 reader 的操作.supports serialize 讀取器 reader 實現是...

caffe讀取多標籤的lmdb資料

問題描述 lmdb檔案支援資料 標籤的形式,但是卻只能寫入乙個標籤,引入多標籤的解決方法有很多,這兒詳細說一下我的辦法 製作多個data資料,分別加入乙個標籤。我的方法只適用於標籤數量較少的情況,標籤數量比較多的話建議修改原始碼支援。下面介紹詳細步驟。以下均以兩個標籤作為介紹。生成兩個含單標籤的li...