caffe中数据库的设计

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;

}