2017-05-12 33 views
1

我有幾個數據庫,我需要在NVIDIA DIGITS上對它們進行分類。但是將我的大數據導入DIGITS需要很長時間(2-4天)! 想象我已經轉換2個圖像集成.lmdb形式,如:如何將外部預編譯的.LMDB文件導入NVIDIA DIGITS?

data1           data2 
--> folder train1_db: data.mdb, lock.mdb  --> folder train2_db: data.mdb, lock.mdb 
--> folder val1_db: data.mdb, lock.mdb   --> folder val2_db: data.mdb, lock.mdb 
--> mean.binaryproto       --> mean.binaryproto 
--> some other txt files...      --> some other txt files... 

現在我需要連接這兩個.lmdb數據庫,並節省時間。所以我已經完成了python從Merge two LMDB databases for feeding to the network (caffe)

我有第三個數據集包含:train_db和val_db文件夾,每個包含data.mdb和lock.mdb文件,如上所述。

data3           
--> folder train3_db: train1_db + train2_db    
--> folder va3_db: val_db + va2_db      

我需要將這些導入到DIGITS中,以便在它們上訓練網絡。

我的問題是:

1 - 我應該導入folders train_db和val_db在image LMDB一部分?

2-我搜索label LMDB但我不明白我應該在這部分做什麼。你能否清楚解釋我應該怎麼做?

非常感謝您的幫助。

回答

1

您必須以與他們相同的方式創建它們。我先讀他們然後創造他們做了什麼。

如果使用相同的類結構更改現有的分類數據集,這將起作用。你必須編輯pickle文件來更新2個地方的train和val的圖像總數。您必須像生成lmdb文件一樣生成lmdb文件。

順便說一句......當然,他們並不推薦這樣做: 退房: https://github.com/NVIDIA/DIGITS/issues/1035

這裏是我的代碼: https://github.com/GemHunt/lmdb-testing/blob/master/create_lmdb_rotate_whole_image.py

+0

感謝。我得到了我的回答:) – azzz

相關問題