caffe python讀取lmdb檔案

2021-08-18 21:49:02 字數 893 閱讀 5022

一般用caffe訓練模型的時候,都會首先把資料寫入資料庫,lmdb/leveldb, 但如果寫入之後想check一下是否寫對了,怎麼去讀lmdb檔案呢?

這裡用python**讀取lmdb檔案,檢視其中的內容,看是否符合預期。

**如下:

-python **

01import caffe

02import lmdb

0304

lmdb_env = lmdb.open('train_lmdb') #caffe儲存lmdb檔案的目錄

05lmdb_txn = lmdb_env.begin()

06lmdb_cursor = lmdb_txn.cursor()

07datum = caffe.proto.caffe_pb2.datum()

0809

for key, value in lmdb_cursor:

10datum.parsefromstring(value)

11label = datum.label

12data = caffe.io.datum_to_array(datum)

13print label, data

這裡匯入了lmdb和caffe兩個庫,解釋一下如何匯入

1、lmdb,需要用pip安裝lmdb,進入root賬戶,輸入 pip install lmdb即可(在其它使用者下,輸入sudo命令無法安裝成功)

2、import caffe, 安裝好caffe後,在bashrc中,新增環境變數, export pythonpyth="***/caffe-master/python", 再source ~/.bashrc

import caffe的時候,會報乙個錯,是因為沒有編譯pycaffe,在caffe-master下,make pycaffe即可。

caffe Python特徵抽取

caffe大家一般用到的深度學習平台都是這個,關於caffe的訓練通常一般都可以通過一些命令來執行,但是在deploy階段,如果是做實際的工程,那麼c 介面用得會相對比較多。但是caffe是支援python和matlab介面的,所以用python來做一些相關的特徵的處理以及額外的任務比較方便 這裡我...

Caffe Python特徵抽取

caffe大家一般用到的深度學習平台都是這個,關於caffe的訓練通常一般都可以通過一些命令來執行,但是在deploy階段,如果是做實際的工程,那麼c 介面用得會相對比較多。但是caffe是支援python和matlab介面的,所以用python來做一些相關的特徵的處理以及額外的任務比較方便 這裡我...

Caffe Python開發環境設定

由於安裝 python 的第三方庫的時候需要編譯,所以需要安裝下面兩個庫 sudo yum install python devel numpy設定virtualenv環境 virtualenv caffeenv cd caffeenv bin activate安裝 python 第三方庫 cd c...