Caffe中数据库在include/util/db.hpp中定义三个接口:Cursor,Transacation,DB。
Cursor 为遍历数据库用的游标,SeekToFirst定位到数据库开始的位置,Next把游标指向下一个位置,key,value方法返回键和值,valid方法检查游标的有效性。
Transaction定义事物控制,Put向数据库中加入对象,Commit提交事务。
DB 为数据库入口,Open打开一个数据库,Close关闭数据库。 NewCursor创建一个新的游标。NewTransacation创建一个新的事务。
GetDB有两种形态,一种是依据枚举量指定后端,一个是用字符串指定后端。后端有leveldb和lmdb。
using namespace caffe; // NOLINT(build/namespaces)
using std::string; using namespace std; int main(int argc, char** argv) {::google::InitGoogleLogging(argv[0]);
auto db = db::GetDB("lmdb");
if (db == nullptr) { printf("Create Database failed\n"); return -1; }db->Open("examples/mnist/mnist_train_lmdb", db::READ);
auto cursor = db->NewCursor();
while (cursor->valid()) { cout << cursor->key() << cursor->value() << endl; } printf("Succes\n");delete db;
return 0;}